File size: 1,065 Bytes
00d99c2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
{
"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
}
|