Add model
Browse files- config.json +4 -0
config.json
CHANGED
|
@@ -6,6 +6,10 @@
|
|
| 6 |
"cropmodel": {
|
| 7 |
"balance_classes": false,
|
| 8 |
"batch_size": 4,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
"lr": 0.0001,
|
| 10 |
"num_classes": 2,
|
| 11 |
"num_workers": 0,
|
|
|
|
| 6 |
"cropmodel": {
|
| 7 |
"balance_classes": false,
|
| 8 |
"batch_size": 4,
|
| 9 |
+
"label_dict": {
|
| 10 |
+
"Alive": 0,
|
| 11 |
+
"Dead": 1
|
| 12 |
+
},
|
| 13 |
"lr": 0.0001,
|
| 14 |
"num_classes": 2,
|
| 15 |
"num_workers": 0,
|