cnn_transfer / seed_132 /README.md
abir-hr196's picture
Add README for seed 132
5c87b03 verified

MNIST CNN Model (Seed 132)

This is a Convolutional Neural Network (CNN) trained on the MNIST dataset with random seed 132.

Run Details

How to Use

import torch

# Load the model
model_path = "seed_132/pytorch_model.bin"
# model = YourCNNClass()
# model.load_state_dict(torch.load(model_path))
# model.eval()