mctr-mmptrack / cafe /train_config.yaml
deepsworld's picture
MCTR MMPTracking finetuned checkpoints (paper reproduction)
60d6f87 verified
Raw History Blame Contribute Delete
6.84 kB
!!python/object/new:ml.utils.config.Config
dictitems:
CMD: train
DATASET: &id008 !!python/object/new:ml.utils.config.Config
dictitems:
AUGUMENTATIONS: true
CAMS: &id003 !!python/object/new:ml.utils.config.Config
dictitems:
cafe_shop: null
industry_safety: null
lobby: null
office: null
retail: null
state:
cafe_shop: null
industry_safety: null
lobby: null
office: null
retail: null
FRAMES_EXT: jpg
FRAMES_PATH: images
FRAMES_PER_SAMPLE: 4
FRAME_RANGES: &id004 !!python/object/new:ml.utils.config.Config
dictitems:
cafe_shop: null
industry_safety: null
lobby: null
office: null
retail: null
state:
cafe_shop: null
industry_safety: null
lobby: null
office: null
retail: null
FRAME_SHAPE: &id005
- 360
- 640
FRAME_STRIDE: 1
LABELS_EXT: json
LABELS_PATH: labels
LOCATIONS: &id006 !!python/object/new:ml.utils.config.Config
dictitems:
train: &id001 !!python/object/new:ml.utils.config.Config
dictitems:
63am: cafe
64am: cafe
state:
63am: cafe
64am: cafe
validation: &id002 !!python/object/new:ml.utils.config.Config
dictitems:
64pm: cafe
state:
64pm: cafe
state:
train: *id001
validation: *id002
NUM_FRAMES: 1
ROOT: /net/mlfs02/data/projects/shared/datasets/MMPTracking
SUBSAMPLE_RATE: 4
state:
AUGUMENTATIONS: true
CAMS: *id003
FRAMES_EXT: jpg
FRAMES_PATH: images
FRAMES_PER_SAMPLE: 4
FRAME_RANGES: *id004
FRAME_SHAPE: *id005
FRAME_STRIDE: 1
LABELS_EXT: json
LABELS_PATH: labels
LOCATIONS: *id006
NUM_FRAMES: 1
ROOT: /net/mlfs02/data/projects/shared/datasets/MMPTracking
SUBSAMPLE_RATE: 4
DEV: &id009 !!python/object/apply:torch.device
- cuda
- 0
DIST: true
DIST_BACKEND: nccl
DIST_NO_SYNC_BN: false
DIST_URL: env://
DRYRUN: false
GPU: 0
INITIALIZE: runs/cafe/finetune-continue-b32-600iter-100epochs-frz_dtrTrue-res_unused3-res_tgtFalse-clplinear-R50-cipr-gpu17_May10_152211/checkpoints/PAIRWISE.pth
LOGDIR: runs/cafe
LOG_PREFIX: finetune-continue-b32-600iter-100epochs
MODELS: &id010 !!python/object/new:ml.utils.config.Config
dictitems:
PAIRWISE: &id007 !!python/object/new:ml.utils.config.Config
dictitems:
ADD_BBOXES: true
BACKBONE: r50
CHECKPOINT: /net/mlfs01/export/users/dpatel/mcmot/checkpoints/detr/detr-r50-e632da11.pth
CLIP_LENGTH_PROGRESSION: linear
CONTEXT_SIZE: 1
DETACHED_UNTIL: 0
EMBEDDING_LOSS: false
FREEZE_DETR: true
FREEZE_DETR_BACKBONE: false
NUM_CLASSES: 1
NUM_QUERIES: 20
PAIRWISE_LOSS: true
PRED_TRACK_BOXES: true
PRED_TRACK_BOXES_TIME: true
PW_LAYERS: 1
PW_NO_PROJ: false
PW_PASSTHROUGH_SRC: true
PW_PROJ_DIM_SCALE: 1
PW_QUERIES: 20
RESET_TGT: false
RESET_TGT_UNTIL: 0
RESET_UNUSED: 3
SCALED_COSTS: true
SELF_ATTENTION_AFTER: true
SELF_ATTENTION_BEFORE: false
TGT_AS_POS: false
WEIGHTED_VIEWS: false
ZERO_QE: true
state:
ADD_BBOXES: true
BACKBONE: r50
CHECKPOINT: /net/mlfs01/export/users/dpatel/mcmot/checkpoints/detr/detr-r50-e632da11.pth
CLIP_LENGTH_PROGRESSION: linear
CONTEXT_SIZE: 1
DETACHED_UNTIL: 0
EMBEDDING_LOSS: false
FREEZE_DETR: true
FREEZE_DETR_BACKBONE: false
NUM_CLASSES: 1
NUM_QUERIES: 20
PAIRWISE_LOSS: true
PRED_TRACK_BOXES: true
PRED_TRACK_BOXES_TIME: true
PW_LAYERS: 1
PW_NO_PROJ: false
PW_PASSTHROUGH_SRC: true
PW_PROJ_DIM_SCALE: 1
PW_QUERIES: 20
RESET_TGT: false
RESET_TGT_UNTIL: 0
RESET_UNUSED: 3
SCALED_COSTS: true
SELF_ATTENTION_AFTER: true
SELF_ATTENTION_BEFORE: false
TGT_AS_POS: false
WEIGHTED_VIEWS: false
ZERO_QE: true
state:
PAIRWISE: *id007
NOTENSORBOARD: false
PREFIX: ''
RANK: 0
RESUME: runs/cafe/finetune-continue-b32-600iter-100epochs-frz_dtrTrue-res_unused3-res_tgtFalse-clplinear-R50-cipr-gpu17_May10_152211/checkpoints/PAIRWISE.pth
SEED: 1204
TRAINING: &id011 !!python/object/new:ml.utils.config.Config
dictitems:
ADAM_EPS: 1.0e-08
AMP: false
BATCH_SIZE: 8
COMMENT: R50
CUDNN_BENCHMARK: false
FIND_UNUSED_PARAM: false
ITS_PER_EPOCH: 600
LOADER_NUM_WORKERS: 8
LR: 0.0001
LR_DECAY_RATIO: 0.3
LR_DETR_MULT: 1
MAX_GRAD_NORM: 3.0
MAX_NUM_EPOCHS: 100
MODEL: PAIRWISE
MOMENTUM: 0.9
NESTEROV: true
OPTIMIZER: adamw
PATIENCE: 0
PIN_MEMORY: true
PRINT_FREQ: 100
SAVE_EVERY: 10
SCHEDULER: step
START_EPOCH: 61
STEP_SIZE: 20
WARMUP_MULTIPLIER: 2
WEIGHT_DECAY: 0.0
state:
ADAM_EPS: 1.0e-08
AMP: false
BATCH_SIZE: 8
COMMENT: R50
CUDNN_BENCHMARK: false
FIND_UNUSED_PARAM: false
ITS_PER_EPOCH: 600
LOADER_NUM_WORKERS: 8
LR: 0.0001
LR_DECAY_RATIO: 0.3
LR_DETR_MULT: 1
MAX_GRAD_NORM: 3.0
MAX_NUM_EPOCHS: 100
MODEL: PAIRWISE
MOMENTUM: 0.9
NESTEROV: true
OPTIMIZER: adamw
PATIENCE: 0
PIN_MEMORY: true
PRINT_FREQ: 100
SAVE_EVERY: 10
SCHEDULER: step
START_EPOCH: 61
STEP_SIZE: 20
WARMUP_MULTIPLIER: 2
WEIGHT_DECAY: 0.0
WORLD_SIZE: 4
__file__: &id012 !!python/object/apply:pathlib.PosixPath
- configs
- train
- pairwise
- pairwise.yaml
state:
CMD: train
DATASET: *id008
DEV: *id009
DIST: true
DIST_BACKEND: nccl
DIST_NO_SYNC_BN: false
DIST_URL: env://
DRYRUN: false
GPU: 0
INITIALIZE: runs/cafe/finetune-continue-b32-600iter-100epochs-frz_dtrTrue-res_unused3-res_tgtFalse-clplinear-R50-cipr-gpu17_May10_152211/checkpoints/PAIRWISE.pth
LOGDIR: runs/cafe
LOG_PREFIX: finetune-continue-b32-600iter-100epochs
MODELS: *id010
NOTENSORBOARD: false
PREFIX: ''
RANK: 0
RESUME: runs/cafe/finetune-continue-b32-600iter-100epochs-frz_dtrTrue-res_unused3-res_tgtFalse-clplinear-R50-cipr-gpu17_May10_152211/checkpoints/PAIRWISE.pth
SEED: 1204
TRAINING: *id011
WORLD_SIZE: 4
__file__: *id012