| { | |
| "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 | |
| } | |