config.json

{
  "add_cross_proj": false,
  "architectures": [
    "BaseModel"
  ],
  "backbone_dim": 1024,
  "data_roots": [
    "DATA/SHREC2017/SHREC2017"
  ],
  "dataset": "SHREC",
  "dataset_list": [
    "SHREC"
  ],
  "depth": 1,
  "depths": 5,
  "dim": 1024,
  "dim_head": 128,
  "dropout1": 0.0,
  "dropout_rate": 0.5,
  "epochs": 300,
  "frames_per_clip": 32,
  "head": 8,
  "heads": 8,
  "hidden_dim": 256,
  "label_smoothing": 0.0,
  "loss_weight": 0.1,
  "lr": 0.1,
  "mlp_dim": 2048,
  "model": "PSTTransformer",
  "model_init": null,
  "model_type": "points",
  "nsamples": 32,
  "num_basis": 16,
  "num_points": 128,
  "num_token": 4,
  "output_dir": "SHREC_PSTTransformer_2_32_0.2_0.1_32_300_one_stage_0.3",
  "pretrain": false,
  "print_freq": 10,
  "probing": false,
  "radius": 0.2,
  "resume": null,
  "seed": 0,
  "spatial_stride": 2,
  "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": "one_stage",
  "transformers_version": "4.17.0",
  "use_info": false,
  "use_multi_dataset": false,
  "warmup_ratio": 0.3,
  "weight_decay": 0.0001,
  "workers": 16
}