Hugging Face's logo

CELL-LAB
/
PIM-resnet-mnist

PyTorch
pim
mnist
visualization
prototype
Model card Files
xet
 Community
PIM-resnet-mnist / README.md
longnew's picture
longnew
Upload README.md with huggingface_hub
6c2eb66 verified
preview code
|
Raw
History Blame Contribute Delete
1.01 kB
metadata 
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.