final_args.json

{
  "batch_size": 16,
  "batch_size_val": 14,
  "epochs": 100,
  "update_freq": 1,
  "save_ckpt_freq": 10,
  "model": "vit_base_patch16_224",
  "tubelet_size": 2,
  "input_size": 224,
  "fc_drop_rate": 0.0,
  "drop": 0.0,
  "attn_drop_rate": 0.0,
  "drop_path": 0.0,
  "disable_eval_during_finetuning": false,
  "model_ema": false,
  "model_ema_decay": 0.9999,
  "model_ema_force_cpu": false,
  "opt": "sgd",
  "opt_eps": 1e-08,
  "opt_betas": null,
  "clip_grad": null,
  "momentum": 0.9,
  "weight_decay": 0.001,
  "weight_decay_end": null,
  "lr": 0.1,
  "layer_decay": 1.0,
  "warmup_lr": 1e-06,
  "min_lr": 1e-06,
  "warmup_epochs": 10,
  "warmup_steps": -1,
  "is_aa": false,
  "color_jitter": 0.4,
  "num_sample": 2,
  "aa": "rand-m7-n4-mstd0.5-inc1",
  "smoothing": 0.1,
  "train_interpolation": "bicubic",
  "crop_pct": null,
  "short_side_size": 224,
  "test_num_segment": 5,
  "test_num_crop": 3,
  "reprob": 0.0,
  "remode": "pixel",
  "recount": 1,
  "resplit": false,
  "mixup": 0.0,
  "cutmix": 0.0,
  "cutmix_minmax": null,
  "mixup_prob": 0.0,
  "mixup_switch_prob": 0.0,
  "mixup_mode": "batch",
  "finetune": "",
  "model_key": "model|module",
  "model_prefix": "",
  "init_scale": 0.001,
  "use_checkpoint": false,
  "use_mean_pooling": true,
  "prompt_num_tokens": 5,
  "prompt_start": null,
  "prompt_end": null,
  "prompt_dropout": 0.0,
  "prompt_init": "random",
  "transfer_type": "prompt",
  "prompt_reparam": false,
  "data_path": "/path/to/list_kinetics-400",
  "eval_data_path": null,
  "nb_classes": 400,
  "imagenet_default_mean_and_std": true,
  "num_segments": 1,
  "num_frames": 16,
  "sampling_rate": 4,
  "data_set": "Kinetics-400",
  "output_dir": "",
  "log_dir": null,
  "device": "cuda",
  "seed": 0,
  "resume": "",
  "auto_resume": true,
  "save_ckpt": true,
  "start_epoch": 0,
  "eval": false,
  "dist_eval": false,
  "num_workers": 10,
  "pin_mem": true,
  "world_size": 3,
  "local_rank": 0,
  "dist_on_itp": false,
  "dist_url": "env://",
  "enable_deepspeed": false,
  "rank": 0,
  "gpu": 0,
  "distributed": true,
  "dist_backend": "nccl"
}