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