FanMingyu commited on
Commit
9daf478
·
verified ·
1 Parent(s): 4a42b43

Add pointtransformer seed 2026 baseline checkpoints

Browse files
pointtransformer/seed_2026/best_loss.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b68a4ede51c42453cde4d6a37273c8201091fd71cfb2234677419ec1dc91446
3
+ size 2914627
pointtransformer/seed_2026/best_pmt_miou.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6dbabe817adccc4d92077fcbf415dfe290c80f74cd2177d5d1689e6287adf12f
3
+ size 2916159
pointtransformer/seed_2026/config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Stage1DirectBaseline"
4
+ ],
5
+ "backbone": "pointtransformer",
6
+ "best_loss": 868.044472501577,
7
+ "best_pmt_miou": 0.3857012391090393,
8
+ "dataset": "ZXCCHENGXI/cstnet2_stage1_mini",
9
+ "epochs": 80,
10
+ "global_batch_size": 128,
11
+ "model_config": {
12
+ "attn_k": 16,
13
+ "attn_neighbors": 20,
14
+ "dgcnn_k": 20,
15
+ "feature_dim": 128,
16
+ "head_dropout": 0.2,
17
+ "head_hidden_dim": 128,
18
+ "model": "pointtransformer",
19
+ "pointmamba_depth": 2,
20
+ "pointmamba_group_size": 32,
21
+ "pointmamba_tokens": 128,
22
+ "pointmamba_width": 64,
23
+ "pointmlp_group_size": 24,
24
+ "pointnext_k": 24,
25
+ "pointtransformer_depth": 3,
26
+ "pointtransformer_k": 16,
27
+ "pointtransformer_width": 64
28
+ },
29
+ "model_type": "cstnet2_stage1_direct_baseline",
30
+ "per_gpu_batch_size": 32,
31
+ "point_count": 2048,
32
+ "precision": "bfloat16",
33
+ "recommended_checkpoint": "best_pmt_miou.pth",
34
+ "seed": 2026,
35
+ "source_code": "https://github.com/xcheng-tsinghua/cstnet2",
36
+ "world_size": 4
37
+ }
pointtransformer/seed_2026/history.json ADDED
The diff for this file is too large to render. See raw diff
 
pointtransformer/seed_2026/last.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a03d9f417290e6212a491c82514d5b46f22f51ceddb59628ff25db5e7db6224b
3
+ size 2912712