You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
765 B
34 lines
765 B
seed: 0
|
|
mixed_precision: true
|
|
base_lr: 4.0e-4
|
|
# base_lr: 0.00001
|
|
t_max: 16100
|
|
|
|
nr_gpus: 3
|
|
batch_size: 3
|
|
n_total_epoch: 100
|
|
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
|
|
# 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
|
|
|
|
|
|
log_level: "logging.INFO"
|
|
|
|
|