Downloads · 30 days
0
walnut-v0-1
mihirneal/walnut-v0-1
walnut-v0-1 is a machine learning model from mihirneal. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for pytorch.
Pretrained 3D ViT-MAE checkpoints for structural MRI.
Access
Public
Updated Jul 24, 2026
Repo size
56.2 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth24.9 GB · 100%
At a glance
- Library
- pytorch
- Access
- Public
- Created
- May 12, 2026
- Updated
- Jul 24, 2026
- SHA
- bdedde38
Model card
From the Hugging Face model README
Walnut v0.1
Pretrained 3D ViT-MAE checkpoints for structural MRI.
The vitb and vitl directories contain models trained with subject-specific
sampling at four dataset scales. These runs used a fixed compute budget of
99,000 optimizer steps with global batch size 256. Each directory contains the
final checkpoint and its training configuration.
vitl/sub-52k/ddep8 is the corresponding full-subject ViT-L experiment with an
eight-layer decoder.
- Library
- pytorch
- Created
- May 12, 2026
- Updated
- Jul 24, 2026