config.json

{
  "add_cross_proj": false,
  "architectures": [
    "BaseModel"
  ],
  "backbone_dim": 2048,
  "data_roots": [
    "DATA/UTD-MHAD/video"
  ],
  "dataset": "UTD",
  "dataset_list": [
    "UTD"
  ],
  "dropout_rate": 0.0,
  "epochs": 20,
  "frames_per_clip": 24,
  "label_smoothing": 0.0,
  "lr": 0.05,
  "meta_list": [
    "DATA/UTD-MHAD/utd.list"
  ],
  "model": "PSTNet",
  "model_init": "zxh4546/pstnet-pde-msr-pretrain",
  "model_type": "points",
  "nsamples": 9,
  "num_points": 2048,
  "output_dir": "UTD_PSTNet_0.05_32_20_encoder_only_init_from_pstnet-pde-msr-pretrain",
  "pretrain": false,
  "print_freq": 10,
  "probing": false,
  "radius": 0.1,
  "seed": 0,
  "step_between_clips": 2,
  "torch_dtype": "float32",
  "train_batch_size": 32,
  "train_mode": "encoder_only",
  "transformers_version": "4.17.0",
  "use_info": false,
  "use_multi_dataset": false,
  "warmup_ratio": 0.1,
  "weight_decay": 0.0001,
  "workers": 16
}