init

config.json

@@ -0,0 +1,58 @@
+{
+  "add_cross_proj": false,
+  "architectures": [
+    "BaseModel"
+  ],
+  "backbone_dim": 1024,
+  "data_roots": [
+    "DATA/SHREC2017/SHREC2017"
+  ],
+  "dataset": "SHREC",
+  "dataset_list": [
+    "SHREC"
+  ],
+  "depth": 1,
+  "depths": 5,
+  "dim": 1024,
+  "dim_head": 128,
+  "dropout1": 0.0,
+  "dropout_rate": 0.5,
+  "epochs": 300,
+  "frames_per_clip": 32,
+  "head": 8,
+  "heads": 8,
+  "hidden_dim": 256,
+  "label_smoothing": 0.0,
+  "loss_weight": 0.1,
+  "lr": 0.1,
+  "mlp_dim": 2048,
+  "model": "PSTTransformer",
+  "model_init": null,
+  "model_type": "points",
+  "nsamples": 32,
+  "num_basis": 16,
+  "num_points": 128,
+  "num_token": 4,
+  "output_dir": "SHREC_PSTTransformer_2_32_0.2_0.1_32_300_one_stage_0.3",
+  "pretrain": false,
+  "print_freq": 10,
+  "probing": false,
+  "radius": 0.2,
+  "resume": null,
+  "seed": 0,
+  "spatial_stride": 2,
+  "start_epoch": 0,
+  "step_between_clips": 1,
+  "tem": 0.1,
+  "temporal_kernel_size": 3,
+  "temporal_stride": 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.3,
+  "weight_decay": 0.0001,
+  "workers": 16
+}

optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d0ffda630ee413c383d2e9fef2da1881e3e7ea9198c2772d2a996c13c87c596f
+size 169189398

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2f3dbf16f10778e91432554c0456e802a92c345199b4146756976c8164736560
+size 169197526

rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a639270e988072513c9c28551c200e71f3c62b2e65ae4931887d5b6e22f7061e
+size 14503

scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ba2343ff25f8daa84b2d04a633c28bb49d2006086ebf3d790a32b928c27206e6
+size 559

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4e973b8c7b1bd22e0a3f7f95ca29af31c5642c2d18e63ef8082060bf4f2b4cc6
+size 3119