output_dir: /results/rgb_3d_ptm encryption_key: nvidia_tao model_config: model_type: rgb backbone: resnet18 rgb_seq_length: 32 input_type: 3d sample_strategy: consecutive dropout_ratio: 0.0 train_config: optim: lr: 0.01 momentum: 0.9 weight_decay: 0.0005 lr_scheduler: MultiStep #lr_steps: [5, 15, 20] lr_steps: [30, 60, 80] lr_decay: 0.1 epochs: 100 checkpoint_interval: 1 dataset_config: train_dataset_dir: /data/train val_dataset_dir: /data/test label_map: #fall_floor: 0 #ride_bike: 1 #clap: 0 #drink: 1 #punch: 2 #push: 3 #run: 4 #shake_hands: 5 #sit: 6 #smoke: 7 #turn: 8 #wave: 9 brush_hair: 0 cartwheel: 1 catch: 2 chew: 3 clap: 4 climb: 5 climb_stairs: 6 dive: 7 draw_sword: 8 dribble: 9 drink: 10 eat: 11 fall_floor: 12 fencing: 13 flic_flac: 14 golf: 15 handstand: 16 hit: 17 hug: 18 jump: 19 kick_ball: 20 kick: 21 kiss: 22 laugh: 23 pick: 24 pour: 25 pullup: 26 punch: 27 push: 28 pushup: 29 ride_bike: 30 ride_horse: 31 run: 32 shake_hands: 33 shoot_ball: 34 shoot_bow: 35 shoot_gun: 36 sit: 37 situp: 38 smile: 39 smoke: 40 somersault: 41 stand: 42 swing_baseball: 43 sword_exercise: 44 sword: 45 talk: 46 throw: 47 turn: 48 walk: 49 wave: 50 output_shape: - 224 - 224 batch_size: 64 workers: 8 clips_per_video: 5 augmentation_config: train_crop_type: random_crop horizontal_flip_prob: 0.5 rgb_input_mean: [0.5] rgb_input_std: [0.5] val_center_crop: True crop_smaller_edge: 256