changed some stuff, re-added default CREStereo scheduler

cfgs/train.yaml

@@ -1,12 +1,12 @@
 seed: 0
-mixed_precision: false
-# base_lr: 4.0e-4
-base_lr: 0.001
-t_max: 161
+mixed_precision: true
+base_lr: 4.0e-4
+# base_lr: 0.00001
+t_max: 16100
 
 nr_gpus: 3
-batch_size: 2
-n_total_epoch: 300
+batch_size: 3
+n_total_epoch: 100
 minibatch_per_epoch: 500
 
 loadmodel: ~
@@ -17,11 +17,18 @@ model_save_freq_epoch: 1
 max_disp: 256
 image_width: 640
 image_height: 480
+# dataset: "blender"
 # training_data_path: "./stereo_trainset/crestereo"
-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"
+# training_data_path: "/media/Data1/connecting_the_dots_data/blender_renders/data"
+training_data_path: "/media/Data1/connecting_the_dots_data/blender_renders_ctd_randomize_light/data"
+
+
+# FIXME any of this??
+pattern_attention: false
+scene_attention: true
+ignore_pattern_completely: false
+
 
 log_level: "logging.INFO"