File size: 1,007 Bytes
6c2eb66 | 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 | ---
license: mit
library_name: pytorch
tags:
- pim
- mnist
- visualization
- prototype
---
# PIM ResNet MNIST Prototype
This model repository stores the prototype checkpoint used by
`CELL-POSTECH/PIM_matmul_visualization`.
## Intended Use
The checkpoint is intended for local interactive visualization of PIM tiled
convolution internals. It is **PROTOTYPE_ONLY** and must not be cited as a
validated benchmark, reproduction, or final MNIST result.
## Files
- `stage9_resnet18_mnist_binary_with_pim_conv2_5_3epoch_prototype_seed0.pt`:
PyTorch `state_dict`.
- `resnet_mnist_3epoch_prototype.json`: configuration used to instantiate the
model and backend visualization slice.
## Prototype Configuration
- Dataset: MNIST
- Epochs: 3
- Seed: 0
- Optimizer: Adam
- Learning rate: 0.0001
- Batch size: 50
- PIM tiled convolution indices: 2, 3, 4, 5
- Result status: `PROTOTYPE_ONLY`
These settings come from the local project prototype config, not from a
paper-verified experimental protocol.
|