Update parameters.json
Browse files- parameters.json +1 -5
parameters.json
CHANGED
|
@@ -9,7 +9,7 @@
|
|
| 9 |
"criterion": "CEL",
|
| 10 |
"optimizer": "Adam",
|
| 11 |
"lr": null,
|
| 12 |
-
"epochs":
|
| 13 |
"batch_size": 64,
|
| 14 |
"bit_depth": 8,
|
| 15 |
"in_channel": 1,
|
|
@@ -33,8 +33,4 @@
|
|
| 33 |
2,
|
| 34 |
3
|
| 35 |
],
|
| 36 |
-
"dataset_info": {
|
| 37 |
-
"train": 301912,
|
| 38 |
-
"val": 37403
|
| 39 |
-
}
|
| 40 |
}
|
|
|
|
| 9 |
"criterion": "CEL",
|
| 10 |
"optimizer": "Adam",
|
| 11 |
"lr": null,
|
| 12 |
+
"epochs": 100,
|
| 13 |
"batch_size": 64,
|
| 14 |
"bit_depth": 8,
|
| 15 |
"in_channel": 1,
|
|
|
|
| 33 |
2,
|
| 34 |
3
|
| 35 |
],
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
}
|