Add MnemoDyn checkpoint

Orion_450_ukbiobank/README.md

@@ -0,0 +1,33 @@
+---
+license: cc-by-nc-4.0
+library_name: pytorch-lightning
+tags:
+  - neuroscience
+  - fmri
+  - time-series
+  - pytorch-lightning
+---
+
+# MnemoDyn Checkpoint
+
+This repository contains a MnemoDyn checkpoint exported from this codebase.
+
+## Checkpoint
+
+- Source checkpoint: `model-epoch=28-val_mae=0.0002.ckpt`
+- Dataset: `GordonHCP`
+- Model class: `coe.light.model.main:LitORionModelOptimized`
+- Remote directory: `Orion_450_ukbiobank`
+
+## Files
+
+- `Orion_450_ukbiobank/model.ckpt`: Lightning checkpoint
+- `Orion_450_ukbiobank/hparams.yaml`: training hyperparameters (if available)
+- `Orion_450_ukbiobank/metrics.csv`: training/validation metrics (if available)
+- `Orion_450_ukbiobank/load_from_hf.py`: minimal loading script
+
+## Usage
+
+```bash
+python Orion_450_ukbiobank/load_from_hf.py --repo_id vhluong/MnemoDyn
+```

Orion_450_ukbiobank/hparams.yaml

@@ -0,0 +1,52 @@
+D: 450
+D_out:
+- 450
+- 450
+- 450
+- 450
+K_LC:
+- 2
+- 2
+- 2
+- 2
+- 2
+- 2
+K_dense: 4
+LC_rank: 4
+conv_bias: true
+dataset_type: GordonHCP
+dense_rank: 8
+dim_d: 100
+dim_k: 300
+duration: 1
+epoch: 1000
+exp: debug
+h_rank: 3
+huber_delta: 1.0
+interpol: spline
+linear_bias: true
+loss_type: total
+lr: 0.001
+masked_modelling: false
+model_pred_save_freq: 100
+n_levels: 5
+nb: 5
+nonlinearity: tanh
+normalize: robust_scaler
+num_classes: 1
+num_sparse_LC: 4
+one_channel: -1
+original_length: 490
+predict: false
+profile: false
+scale: false
+seed: 2025
+storage: /mnt/vhluong/Result/Orion_450_ukbiobank/
+subset: true
+test_bs: 16
+train_bs: 8
+use_mRLoss: false
+use_mixed_precision: false
+use_residual: true
+valid_bs: 16
+wave: db2

Orion_450_ukbiobank/load_from_hf.py

@@ -0,0 +1,20 @@
+#!/usr/bin/env python3
+import argparse
+from huggingface_hub import hf_hub_download
+from coe.light.model.main import LitORionModelOptimized
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--repo_id", required=True, help="e.g. your-name/mnemodyn-gordon333")
+    parser.add_argument("--revision", default="main")
+    args = parser.parse_args()
+
+    ckpt_path = hf_hub_download(repo_id=args.repo_id, filename="Orion_450_ukbiobank/model.ckpt", revision=args.revision)
+    model = LitORionModelOptimized.load_from_checkpoint(ckpt_path, map_location="cpu")
+    model.eval()
+    print("Loaded model from", ckpt_path)
+
+
+if __name__ == "__main__":
+    main()

Orion_450_ukbiobank/metrics.csv

@@ -0,0 +1,65 @@
+epoch,step,train_loss,train_mae,train_mse,val_loss,val_mae,val_mse
+0,282,,,,14.926526069641113,2.7690324783325195,12.157491683959961
+0,282,16.18577003479004,2.8943538665771484,13.291399955749512,,,
+1,565,,,,14.142614364624023,2.687645196914673,11.454971313476562
+1,565,14.449384689331055,2.7200140953063965,11.72937297821045,,,
+2,848,,,,13.697771072387695,2.640517234802246,11.057251930236816
+2,848,13.895867347717285,2.661860704421997,11.233997344970703,,,
+3,1131,,,,13.368184089660645,2.605036973953247,10.763152122497559
+3,1131,13.53722095489502,2.6235547065734863,10.91366958618164,,,
+4,1414,,,,12.939892768859863,2.5583810806274414,10.381510734558105
+4,1414,13.143905639648438,2.5809288024902344,10.562979698181152,,,
+5,1697,,,,12.667008399963379,2.528226852416992,10.138781547546387
+5,1697,12.78211498260498,2.5412373542785645,10.240880966186523,,,
+6,1980,,,,12.444743156433105,2.5034096240997314,9.941335678100586
+6,1980,12.543871879577637,2.5147836208343506,10.029086112976074,,,
+7,2263,,,,12.313505172729492,2.488647222518921,9.824856758117676
+7,2263,12.378240585327148,2.496225595474243,9.882022857666016,,,
+8,2546,,,,12.241231918334961,2.4805119037628174,9.760720252990723
+8,2546,12.273860931396484,2.4844672679901123,9.789398193359375,,,
+9,2829,,,,12.210455894470215,2.4770121574401855,9.733443260192871
+9,2829,12.220453262329102,2.4784247875213623,9.742020606994629,,,
+10,3112,,,,11.798908233642578,2.430083751678467,9.368825912475586
+10,3112,11.99897575378418,2.4532649517059326,9.545711517333984,,,
+11,3395,,,,11.712701797485352,2.42014479637146,9.292552947998047
+11,3395,11.730295181274414,2.422499179840088,9.307796478271484,,,
+12,3678,,,,11.812843322753906,2.4316799640655518,9.3811674118042
+12,3678,11.761422157287598,2.4260361194610596,9.335392951965332,,,
+13,3961,,,,11.755184173583984,2.4250590801239014,9.33012580871582
+13,3961,11.80979061126709,2.4316349029541016,9.37814998626709,,,
+14,4244,,,,10.472766876220703,2.273037910461426,8.199727058410645
+14,4244,11.221672058105469,2.3627657890319824,8.858906745910645,,,
+15,4527,,,,8.160612106323242,1.9732768535614014,6.187336444854736
+15,4527,9.374866485595703,2.1333706378936768,7.241490364074707,,,
+16,4810,,,,7.326295375823975,1.8546894788742065,5.47160530090332
+16,4810,7.700925350189209,1.9087026119232178,5.7922282218933105,,,
+17,5093,,,,6.650959014892578,1.7535446882247925,4.897415637969971
+17,5093,7.0145583152771,1.8086563348770142,5.205893516540527,,,
+18,5376,,,,5.861040115356445,1.6285243034362793,4.232515335083008
+18,5376,6.276676654815674,1.695176124572754,4.581501483917236,,,
+19,5659,,,,5.439999103546143,1.5584803819656372,3.8815176486968994
+19,5659,5.59626579284668,1.5849210023880005,4.011345386505127,,,
+20,5942,,,,4.739255905151367,1.4357736110687256,3.3034825325012207
+20,5942,5.177693843841553,1.513469934463501,3.6642239093780518,,,
+21,6225,,,,3.1591343879699707,1.119905710220337,2.039228916168213
+21,6225,3.8627402782440186,1.2662262916564941,2.5965166091918945,,,
+22,6508,,,,1.6918176412582397,0.7456679940223694,0.9461498260498047
+22,6508,2.5327768325805664,0.9693958163261414,1.5633816719055176,,,
+23,6791,,,,0.12547928094863892,0.09991554915904999,0.02556367591023445
+23,6791,0.7164803147315979,0.37956321239471436,0.3369174599647522,,,
+24,7074,,,,0.010128469206392765,0.009494948200881481,0.0006335200159810483
+24,7074,0.046961311250925064,0.04021431505680084,0.006747000850737095,,,
+25,7357,,,,0.0017157017719000578,0.0016586772399023175,5.7024724810617045e-05
+25,7357,0.004322184715420008,0.004104132764041424,0.00021804560674354434,,,
+26,7640,,,,0.000841438421048224,0.00081069819862023,3.074065534747206e-05
+26,7640,0.0010402201442047954,0.0010029561817646027,3.726364593603648e-05,,,
+27,7923,,,,0.00042236794251948595,0.00039254772127605975,2.9820170311722904e-05
+27,7923,0.0005542469443753362,0.0005251383408904076,2.9108714443282224e-05,,,
+28,8206,,,,0.0002644442138262093,0.0002342492516618222,3.0194863938959315e-05
+28,8206,0.00028470935649238527,0.0002555803512223065,2.9128952519386075e-05,,,
+29,8489,,,,0.00018961283785756677,0.00015914843243081123,3.046440178877674e-05
+29,8489,0.00019706755119841546,0.00016763397434260696,2.943363506346941e-05,,,
+30,8772,,,,0.007168771233409643,0.007060167379677296,0.00010860298061743379
+30,8772,0.006942206993699074,0.006825406104326248,0.00011679960152832791,,,
+31,9055,,,,0.010919344611465931,0.010706053115427494,0.0002132911467924714
+31,9055,0.00848153792321682,0.008328153751790524,0.00015338619414251298,,,

Orion_450_ukbiobank/model.ckpt

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