diff --git a/cfgs/train.yaml b/cfgs/train.yaml index cfc5f7b..fc54fa2 100644 --- a/cfgs/train.yaml +++ b/cfgs/train.yaml @@ -2,7 +2,7 @@ seed: 0 mixed_precision: false base_lr: 4.0e-4 -nr_gpus: 8 +nr_gpus: 3 batch_size: 4 n_total_epoch: 600 minibatch_per_epoch: 500 @@ -12,9 +12,10 @@ log_dir: "./train_log" model_save_freq_epoch: 1 max_disp: 256 -image_width: 512 -image_height: 384 -training_data_path: "./stereo_trainset/crestereo" +image_width: 640 +image_height: 480 +# training_data_path: "./stereo_trainset/crestereo" +training_data_path: "/media/Data1/connecting_the_dots_data/ctd_data/" log_level: "logging.INFO"