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.

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support