BrainFM / cfgs /trainer /train /joint.yaml
peirong26's picture
Upload 187 files
2571f24 verified
raw
history blame contribute delete
659 Bytes
## job specific set ups ##
exp_name: wosr_reggrad #wosr_reggrad #hemis #wosr_reggrad #wosr_reggrad_lowres # age_pool
job_name: l6_16
init_method: "tcp://localhost:9998"
eval_only: False
debug: False
resume: True
reset_epoch: False
resume_optim: True
ckp_path: /autofs/vast/lemon/temp_stuff/peirong/results/MTBrainID/wosr_reggrad/l6_16/1128-0936/ckp/checkpoint_latest.pth
n_epochs: 5000
#lr_drops: [1600] # [70, 90] # [120] # [70, 90]
lr_drops: [2500] # [70, 90] # [120] # [70, 90]
#lr_drops: [660] # [70, 90] # [120] # [70, 90]
condition: #mask+flip # mask, flip, mask+flip
log_itr: 200
vis_itr: 6000
num_levels: 6 # 5 (1024), 6 (2048), 7 (4096)