File size: 937 Bytes
10a6b96 |
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 |
{
"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
}
|