| { | |
| "add_cross_proj": false, | |
| "architectures": [ | |
| "BaseModel" | |
| ], | |
| "backbone_dim": 1024, | |
| "data_roots": [ | |
| "DATA/SHREC2017/SHREC2017" | |
| ], | |
| "dataset": "SHREC", | |
| "dataset_list": [ | |
| "SHREC" | |
| ], | |
| "depths": 5, | |
| "dim": 1024, | |
| "dim_head": 128, | |
| "dropout1": 0.0, | |
| "dropout_rate": 0.5, | |
| "epochs": 300, | |
| "frames_per_clip": 32, | |
| "heads": 8, | |
| "label_smoothing": 0.0, | |
| "lr": 0.1, | |
| "mlp_dim": 2048, | |
| "model": "PSTTransformer", | |
| "model_init": null, | |
| "model_type": "points", | |
| "nsamples": 32, | |
| "num_points": 128, | |
| "output_dir": "SHREC_PSTTransformer_0.1_32_300_encoder_only", | |
| "pretrain": false, | |
| "print_freq": 10, | |
| "probing": false, | |
| "radius": 0.2, | |
| "resume": null, | |
| "seed": 0, | |
| "spatial_stride": 2, | |
| "start_epoch": 0, | |
| "step_between_clips": 1, | |
| "temporal_kernel_size": 3, | |
| "temporal_stride": 1, | |
| "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 | |
| } | |