zxcvb6958 commited on
Commit
0801a47
·
1 Parent(s): e3ed634

Remove standalone Task4 checkpoint metadata

Browse files
task4/neurostorm_hcpya_fmri_reid.metadata.json DELETED
@@ -1,88 +0,0 @@
1
- {
2
- "schema_version": 1,
3
- "model": {
4
- "name": "NeuroSTORM",
5
- "framework": "PyTorch Lightning",
6
- "pytorch_lightning_version": "1.9.4",
7
- "adaptation": "full_finetuning"
8
- },
9
- "task": {
10
- "id": 4,
11
- "name": "fmri_reid",
12
- "description": "HCP-YA fMRI re-identification",
13
- "type": "classification",
14
- "dataset": "HCP1200",
15
- "num_classes": 867
16
- },
17
- "checkpoint": {
18
- "file": "neurostorm_hcpya_fmri_reid.ckpt",
19
- "format": "PyTorch Lightning checkpoint",
20
- "size_bytes": 78485797,
21
- "sha256": "12ad539b6e760adf55274f91eb3a66cf08cb710549a16f77f7548278016369c0",
22
- "epoch": 27,
23
- "global_step": 105700,
24
- "state_dict_tensor_count": 227,
25
- "state_dict_numel": 6516399,
26
- "selection": {
27
- "split": "validation",
28
- "metric": "reid_top1",
29
- "mode": "max",
30
- "value": 0.9980000257492065
31
- }
32
- },
33
- "architecture": {
34
- "data_type": "voxel",
35
- "sequence_length": 40,
36
- "image_size": [96, 96, 96, 40],
37
- "embed_dim": 36,
38
- "depths": [2, 2, 6, 2],
39
- "num_heads": [3, 6, 12, 24],
40
- "c_multiplier": 2,
41
- "last_layer_full_msa": true,
42
- "classifier_head": "v2"
43
- },
44
- "training": {
45
- "pretraining_checkpoint": "pt_neurostorm_mae_5ds_seq40.ckpt",
46
- "optimizer": "AdamW",
47
- "learning_rate": 0.00005,
48
- "weight_decay": 0.01,
49
- "scheduler": true,
50
- "gradient_clipping": true,
51
- "batch_size_per_device": 6,
52
- "devices": 4,
53
- "gradient_accumulation_steps": 1,
54
- "effective_batch_size": 24,
55
- "max_epochs": 30,
56
- "seed": 1234,
57
- "train_fraction": 0.9,
58
- "validation_fraction": 0.1,
59
- "stride_between_sequences": 1,
60
- "stride_within_sequences": 1
61
- },
62
- "evaluation": {
63
- "date": "2026-08-16",
64
- "mode": "single_gpu",
65
- "precision": 32,
66
- "batch_size": 6,
67
- "reid_protocol": {
68
- "gallery_size_identities": 100,
69
- "gallery_size_scans": 200,
70
- "num_trials": 5,
71
- "seed": 1234,
72
- "pseudo_scans": 0,
73
- "eligible_subjects": 105,
74
- "test_scans": 428
75
- },
76
- "test_metrics": {
77
- "reid_top1": 0.9900000095367432,
78
- "reid_top1_std": 0.0077459667809307575,
79
- "reid_mAP": 0.9774730205535889,
80
- "reid_mAP_std": 0.006768760737031698
81
- }
82
- },
83
- "provenance": {
84
- "source_run": "neurostorm_full_seq40_lr5e5_wd001_seed1234_cont140_g1345",
85
- "checkpoint_selection": "Best validation reid_top1 checkpoint",
86
- "test_run": "best_test_epoch27_singlegpu_20260816"
87
- }
88
- }