config.json

{
  "add_cross_proj": false,
  "architectures": [
    "BaseModel"
  ],
  "backbone_dim": 1024,
  "data_roots": [
    "DATA/MSR-Action3D/video"
  ],
  "dataset": "MSRACTION3D",
  "dataset_list": [
    "MSRACTION3D"
  ],
  "depth": 1,
  "depths": 5,
  "dim": 1024,
  "dim_head": 128,
  "dropout1": 0.0,
  "dropout_rate": 0.0,
  "epochs": 45,
  "frames_per_clip": 24,
  "head": 8,
  "heads": 8,
  "hidden_dim": 1024,
  "label_smoothing": 0.0,
  "loss_weight": 0.1,
  "lr": 0.05,
  "meta_list": [
    "DATA/MSR-Action3D/msr.list"
  ],
  "mlp_dim": 2048,
  "model": "PSTTransformer",
  "model_init": null,
  "model_type": "points",
  "nsamples": 32,
  "num_basis": 16,
  "num_points": 2048,
  "num_token": 4,
  "output_dir": "MSRACTION3D_PSTTransformer_0.05_0.3_32_45_two_stage-1024-8-16-4-0.1-0.1",
  "pretrain": true,
  "print_freq": 10,
  "radius": 0.3,
  "resume": null,
  "seed": 0,
  "spatial_stride": 32,
  "start_epoch": 0,
  "step_between_clips": 1,
  "tem": 0.1,
  "temporal_kernel_size": 3,
  "temporal_stride": 1,
  "torch_dtype": "float32",
  "train_batch_size": 32,
  "train_mode": "two_stage",
  "transformers_version": "4.17.0",
  "use_info": false,
  "use_multi_dataset": false,
  "warmup_ratio": 0.1,
  "weight_decay": 0.0001,
  "workers": 16
}