| COMMENT: 'Training on MatterportLayout' | |
| VAL_NAME: 'test' | |
| TAG: 'mp3d_scheduler' | |
| SEED: 123 | |
| MODEL: | |
| NAME: 'LGT_Net' | |
| ARGS: [ { | |
| 'decoder_name': 'SWG_Transformer', | |
| 'win_size': 16, | |
| 'rpe': 'lr_parameter_mirror', | |
| 'dropout': 0.0, | |
| 'depth': 8, | |
| 'output_name': 'LGT' | |
| } ] | |
| TRAIN: | |
| DEVICE: 'cuda:0' | |
| SCRATCH: False | |
| DETERMINISTIC: True | |
| CRITERION: | |
| DEPTH: | |
| WEIGHT: 0.9 | |
| RATIO: | |
| WEIGHT: 0.1 | |
| GRAD: | |
| WEIGHT: 0.1 | |
| WEIGHTS: [ 1.0, 1.0 ] | |
| BASE_LR: | |
| 3e-4 | |
| RESUME_LAST: False | |
| OPTIMIZER: | |
| NAME: 'adam' | |
| EPOCHS: 1000 | |
| LR_SCHEDULER: | |
| NAME: 'StepLR' | |
| ARGS: [ { 'step_size': 20, 'gamma': 0.9, 'last_epoch': -1} ] | |
| DATA: | |
| DATASET: 'mp3d' | |
| DIR: 'src/dataset/mp3d' | |
| BATCH_SIZE: 6 | |
| NUM_WORKERS: 6 | |
| FOR_TEST_INDEX: None | |
| AUG: | |
| FLIP: True | |
| STRETCH: True | |
| ROTATE: True | |
| GAMMA: True | |
| AMP_OPT_LEVEL: 'O0' | |