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 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|