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

change a bunch of stuff, add wip lightning implementation

Browse Source
This commit is contained in:
Cpt.Captain
2022-08-24 16:25:12 +02:00
parent 11959eef61
commit 63da24f429
10 changed files with 916 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cfgs/train.yaml
+6 -2
View File
@@ -4,18 +4,22 @@ base_lr: 4.0e-4
nr_gpus: 3
batch_size: 4
n_total_epoch: 600
n_total_epoch: 300
minibatch_per_epoch: 500
loadmodel: ~
log_dir: "./train_log"
log_dir_lightning: "./train_log_lightning"
model_save_freq_epoch: 1
max_disp: 256
image_width: 640
image_height: 480
# training_data_path: "./stereo_trainset/crestereo"
training_data_path: "/media/Data1/connecting_the_dots_data/ctd_data/"
pattern_attention: true
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"
log_level: "logging.INFO"