File size: 736 Bytes
88b0dcb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
COMMENT: 'Training on MatterportLayout'
TAG: 'mp3d'
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:
1e-4
RESUME_LAST: False
OPTIMIZER:
NAME: 'adam'
EPOCHS: 1000
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'
|