Spaces:
Runtime error
Runtime error
paths: | |
ROOT_OUT_PATH: './results/' | |
ROOT_CHECKPOINT_PATH: './checkpoint/' | |
MODELPATH_NORMFLOW: './checkpoint/barc_normflow_pret/rgbddog_v3_model.pt' | |
smal: | |
SMAL_MODEL_TYPE: '39dogs_norm_newv3' # '39dogs_norm' # '39dogs_diffsize' # 'barc' | |
SMAL_KEYP_CONF: 'olive' # 'green' | |
optim: | |
LR: 5e-5 # 5e-7 # (new) 5e-6 # 5e-5 # 5e-5 # 5e-4 | |
SCHEDULE: [150, 175, 200] # [220, 270] # [150, 175, 200] | |
GAMMA: 0.1 | |
MOMENTUM: 0 | |
WEIGHT_DECAY: 0 | |
EPOCHS: 220 # 300 | |
BATCH_SIZE: 14 # 12 # keep 12 (needs to be an even number, as we have a custom data sampler) | |
TRAIN_PARTS: 'refinement_model' # 'refinement_model_and_shape' # 'refinement_model' | |
params: | |
REF_NET_TYPE: 'multrot01all_res34' # 'multrot01all_res34' # 'multrot01all' # 'multrot01' # 'multrot01' # 'multrot01' # 'multrot' # 'multrot_res34' # 'multrot' # 'add' | |
REF_DETACH_SHAPE: True | |
GRAPHCNN_TYPE: 'multistage_simple' # 'inexistent' | |
SHAPEREF_TYPE: 'inexistent' # 'linear' # 'inexistent' | |
ISFLAT_TYPE: 'linear' # 'inexistent' # 'inexistent' | |
data: | |
DATASET: 'stanext24_withgc_csaddnonflatmorestanding' # 'stanext24_withgc_csaddnonflat' # 'stanext24_withgc_cs0' | |
SHORTEN_VAL_DATASET_TO: 600 # this is faster as we do not evaluate on the whole validation set | |
VAL_OPT: 'val' |