Cpt.Captain/CREStereo-pytorch-nxt
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

adjust config

Browse Source
This commit is contained in:
Nils Koch 2022-05-30 16:23:47 +02:00
parent 70e4bf6fe1
commit 17bf30fa2a
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
cfgs/train.yaml
View File

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