finish lightningification\n\nTraining still seems borked

2022-08-24 19:18:20 +02:00
parent 0e2a4b2340
commit d8169e01bc
4 changed files with 103 additions and 185 deletions
@@ -3,7 +3,7 @@ mixed_precision: false
 base_lr: 4.0e-4

 nr_gpus: 3
-batch_size: 4
+batch_size: 2
 n_total_epoch: 300
 minibatch_per_epoch: 500