debug: True

output_dir: /home/[username]/bimanual_flow/output
model_suffix: '' # suffix to output folder name

dataset_path: /data # dataset root folder
run_name: biman_towel_actpickplace_n10000_h2_co1_am0.9_tc0.5_cam0.65 # train buffer
test_name: biman_towel_actpickplace_n1000_h2_co1_am0.9_tc0.5_cam0.65 # validation buffer

seed: 0 # random seed
epochs: 150 # max training epochs 
log_freq: 500  # prints and logs every frequency steps
eval_freq: 5000 # run validation every frequency steps
save_freq: 5000 # save models and curves frequency
max_buf: 20000 # 7200 # max training buf
im_width: 200 # obs image width
single: False # datset single flag # debug only
loss_type: sigmoid # sigmoid # softmax # softargmax
min_loss: True # use min of the two pick pts
model_type: split # split # 2ch # 1ch # 4ch
swapping: False # also train on swapped pick pts during each train batch
flow_align: False
augment: False # dataset spatial aug flag
rand_depth_offset: False # dataset spatial aug flag
gaussian_noise: False
misalign: False
batch: 10 # batch size # default: 10
lr: 0.0001 # learning rate
pick: True
flow: epoch=613.ckpt #epoch=613.ckpt #rect_epoch=872.ckpt #tsh_epoch=683.ckpt #lact_epoch=513.ckpt
# weight_decay: 0.0001
input_mode: flowonly #noflow #obsflow #flowonly
