@@ -15,7 +15,7 @@ CheckDataset:
val_percent: null
Train:
- num_classes: 4
+ num_classes: 1
epochs_iters: 50
batch_size: 4
learning_rate: 0.08
batch_size: 2
learning_rate: 0.001
learning_rate: 0.4
+ num_classes: 10
learning_rate: 0.01
epochs_iters: 10
learning_rate: 0.00125