adjust config

3 years ago · 17bf30fa2a
parent 70e4bf6fe1
commit 17bf30fa2a
1 changed files with 5 additions and 4 deletions
--- 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"