added other hyperparameters
Browse files
app.py
CHANGED
|
@@ -32,19 +32,43 @@ lw = 3
|
|
| 32 |
ps = 200
|
| 33 |
cmap = 'magma'
|
| 34 |
|
| 35 |
-
effnet_hparams = {
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
# effnet_config = SimpleNamespace(**effnet_hparams)
|
| 49 |
|
| 50 |
# which layers to look at
|
|
|
|
| 32 |
ps = 200
|
| 33 |
cmap = 'magma'
|
| 34 |
|
| 35 |
+
effnet_hparams = {47: {"num_classes": 2,
|
| 36 |
+
"gamma": 0.04294256770072906,
|
| 37 |
+
"lr": 0.010208864616781627,
|
| 38 |
+
"weight_decay": 0.00014537466483781656,
|
| 39 |
+
"batch_size": 16,
|
| 40 |
+
"num_channels": 47,
|
| 41 |
+
"stochastic_depth_prob": 0.017760418815821067,
|
| 42 |
+
"dropout": 0.039061686292663655,
|
| 43 |
+
"width_mult": 0.7540060155156922,
|
| 44 |
+
"depth_mult": 0.9378692812212488,
|
| 45 |
+
"size": "v2_s"},
|
| 46 |
+
61: {
|
| 47 |
+
"num_classes": 2,
|
| 48 |
+
"gamma": 0.032606396652426956,
|
| 49 |
+
"lr": 0.008692971067922545,
|
| 50 |
+
"weight_decay": 0.00008348389688708425,
|
| 51 |
+
"batch_size": 23,
|
| 52 |
+
"num_channels": 61,
|
| 53 |
+
"stochastic_depth_prob": 0.003581930052432713,
|
| 54 |
+
"dropout": 0.027804120950575217,
|
| 55 |
+
"width_mult": 1.060782511229692,
|
| 56 |
+
"depth_mult": 0.7752918857163054,
|
| 57 |
+
"size": "v2_s"},
|
| 58 |
+
75: {
|
| 59 |
+
"num_classes": 2,
|
| 60 |
+
"gamma": 0.029768470449465057,
|
| 61 |
+
"lr": 0.008383851744497892,
|
| 62 |
+
"weight_decay": 0.000196304392793202,
|
| 63 |
+
"batch_size": 32,
|
| 64 |
+
"num_channels": 75,
|
| 65 |
+
"stochastic_depth_prob": 0.08398410137077088,
|
| 66 |
+
"dropout": 0.03351826828687193,
|
| 67 |
+
"width_mult": 1.144132674734038,
|
| 68 |
+
"depth_mult": 1.2267023928285563,
|
| 69 |
+
"size": "v2_s"
|
| 70 |
+
}
|
| 71 |
+
}
|
| 72 |
# effnet_config = SimpleNamespace(**effnet_hparams)
|
| 73 |
|
| 74 |
# which layers to look at
|