File size: 1,082 Bytes
c31b628 |
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 53 |
{
"add_cross_proj": false,
"architectures": [
"BaseModel"
],
"backbone_dim": 2048,
"data_roots": [
"DATA/MSR-Action3D/video"
],
"dataset": "MSRACTION3D",
"dataset_list": [
"MSRACTION3D"
],
"depth": 1,
"dropout_rate": 0.0,
"epochs": 45,
"frames_per_clip": 24,
"head": 8,
"hidden_dim": 256,
"label_smoothing": 0.0,
"loss_weight": 0.1,
"lr": 0.05,
"meta_list": [
"DATA/MSR-Action3D/msr.list"
],
"model": "PSTNet",
"model_init": null,
"model_type": "points",
"nsamples": 9,
"num_basis": 16,
"num_points": 2048,
"num_token": 4,
"output_dir": "MSRACTION3D_PSTNet_0.05_0.3_32_45_one_stage-256-8-16-4-0.1-0.1",
"pretrain": false,
"print_freq": 10,
"probing": false,
"radius": 0.3,
"resume": null,
"seed": 0,
"start_epoch": 0,
"step_between_clips": 1,
"tem": 0.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.1,
"weight_decay": 0.0001,
"workers": 16
}
|