cnn_transfer / seed_112 /README.md
abir-hr196's picture
Add README for seed 112
42c4b22 verified

MNIST CNN Model (Seed 112)

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

Run Details

How to Use

import torch

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