last batch of live-fixes and improvments

This commit is contained in:
2022-09-23 11:27:04 +02:00
parent 2731ef1ada
commit 6f6ac23175
6 changed files with 264 additions and 137 deletions
+5 -13
View File
@@ -1,12 +1,12 @@
seed: 0
mixed_precision: true
base_lr: 4.0e-4
# base_lr: 0.00001
base_lr: 0.00025
t_max: 16100
scheduler: "cosineannealing"
nr_gpus: 3
batch_size: 3
n_total_epoch: 100
n_total_epoch: 64
minibatch_per_epoch: 500
loadmodel: ~
@@ -17,17 +17,9 @@ model_save_freq_epoch: 1
max_disp: 256
image_width: 640
image_height: 480
# dataset: "blender"
# training_data_path: "./stereo_trainset/crestereo"
# 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_ctd_randomize_light/data"
# FIXME any of this??
pattern_attention: false
scene_attention: true
ignore_pattern_completely: false
test_data_path: "./eval_kinect"
data_limit: 1.
log_level: "logging.INFO"