fix lightning, prepare sweeps

2022-08-27 11:21:00 +02:00
parent d8169e01bc
commit 37c537ca31
5 changed files with 151 additions and 87 deletions
@@ -1,6 +1,8 @@
 seed: 0
 mixed_precision: false
-base_lr: 4.0e-4
+# base_lr: 4.0e-4
+base_lr: 0.001
+t_max: 161

 nr_gpus: 3
 batch_size: 2
@@ -16,7 +18,7 @@ max_disp: 256
 image_width: 640
 image_height: 480
 # training_data_path: "./stereo_trainset/crestereo"
-pattern_attention: true
+pattern_attention: false
 dataset: "blender"
 # training_data_path: "/media/Data1/connecting_the_dots_data/ctd_data/"
 training_data_path: "/media/Data1/connecting_the_dots_data/blender_renders/data"