Zhuoxu Huang
commited on
Commit
•
00d99c2
1
Parent(s):
5c2b334
init
Browse files- config.json +51 -0
- optimizer.pt +3 -0
- pytorch_model.bin +3 -0
- rng_state.pth +3 -0
- scheduler.pt +3 -0
- trainer_state.json +0 -0
- training_args.bin +3 -0
config.json
ADDED
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_cross_proj": false,
|
3 |
+
"architectures": [
|
4 |
+
"BaseModel"
|
5 |
+
],
|
6 |
+
"backbone_dim": 1024,
|
7 |
+
"data_roots": [
|
8 |
+
"DATA/SHREC2017/SHREC2017"
|
9 |
+
],
|
10 |
+
"dataset": "SHREC",
|
11 |
+
"dataset_list": [
|
12 |
+
"SHREC"
|
13 |
+
],
|
14 |
+
"depths": 5,
|
15 |
+
"dim": 1024,
|
16 |
+
"dim_head": 128,
|
17 |
+
"dropout1": 0.0,
|
18 |
+
"dropout_rate": 0.5,
|
19 |
+
"epochs": 300,
|
20 |
+
"frames_per_clip": 32,
|
21 |
+
"heads": 8,
|
22 |
+
"label_smoothing": 0.0,
|
23 |
+
"lr": 0.1,
|
24 |
+
"mlp_dim": 2048,
|
25 |
+
"model": "PSTTransformer",
|
26 |
+
"model_init": null,
|
27 |
+
"model_type": "points",
|
28 |
+
"nsamples": 32,
|
29 |
+
"num_points": 128,
|
30 |
+
"output_dir": "SHREC_PSTTransformer_0.1_32_300_encoder_only",
|
31 |
+
"pretrain": false,
|
32 |
+
"print_freq": 10,
|
33 |
+
"probing": false,
|
34 |
+
"radius": 0.2,
|
35 |
+
"resume": null,
|
36 |
+
"seed": 0,
|
37 |
+
"spatial_stride": 2,
|
38 |
+
"start_epoch": 0,
|
39 |
+
"step_between_clips": 1,
|
40 |
+
"temporal_kernel_size": 3,
|
41 |
+
"temporal_stride": 1,
|
42 |
+
"torch_dtype": "float32",
|
43 |
+
"train_batch_size": 32,
|
44 |
+
"train_mode": "encoder_only",
|
45 |
+
"transformers_version": "4.17.0",
|
46 |
+
"use_info": false,
|
47 |
+
"use_multi_dataset": false,
|
48 |
+
"warmup_ratio": 0.1,
|
49 |
+
"weight_decay": 0.0001,
|
50 |
+
"workers": 16
|
51 |
+
}
|
optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a7afd1edf4968e761421f3c666454d9b7c215b14471c7eef8060b2d2b59b650a
|
3 |
+
size 168092712
|
pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fbeee1bd70fe2c064c66de6b0ebdec2c1f0b1bce2da97940855f20f9310415f7
|
3 |
+
size 168100456
|
rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8906817226bcdd62b62416c1d8b4fa4e2a0295d14d494f0a88f62ed29f5a4529
|
3 |
+
size 14503
|
scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3a7f8cc7fe2c21c0d1bbb8fc69367e7915c99302b5a245a2ade99c7268e90337
|
3 |
+
size 559
|
trainer_state.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cd3341fb602a0826589ead5eec95d13ce5f14ce9d8d3b35e35886b5ced3cbaba
|
3 |
+
size 3055
|