File size: 1,363 Bytes
3ed53c2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
---
license: apache-2.0
tags:
- segmentation
- mamba
- wafer
- electron-microscopy
- tokenunify
---
# TokenUnify Models
This repository contains TokenUnify models of different sizes trained on wafer electron microscopy data, along with a superhuman baseline model.
## Available Models
- **TokenUnify-1B.pth**: 1B parameter TokenUnify model
- **TokenUnify-500M.pth**: 500M parameter TokenUnify model
- **TokenUnify-200M.pth**: 200M parameter TokenUnify model
- **TokenUnify-100M.pth**: 100M parameter TokenUnify model
- **superhuman.pth**: Superhuman baseline model
## Model Details
- **Architecture**: TokenUnify (based on Mamba)
- **Training Data**: Wafer electron microscopy images
- **Task**: Image Segmentation
- **Framework**: PyTorch
## Usage
```python
import torch
# Load a specific model
model_path = "TokenUnify-1B.pth" # or any other model file
checkpoint = torch.load(model_path, map_location='cpu')
# Your model loading code here
```
## Model Sizes
| Model | Parameters | File Name |
|-------|------------|-----------|
| TokenUnify Large | 1B | TokenUnify-1B.pth |
| TokenUnify Medium | 500M | TokenUnify-500M.pth |
| TokenUnify Small | 200M | TokenUnify-200M.pth |
| TokenUnify Tiny | 100M | TokenUnify-100M.pth |
| Superhuman Baseline | - | superhuman.pth |
## Citation
If you use these models, please cite the relevant paper.
EOF
|