cnn_transfer / seed_126 /README.md
abir-hr196's picture
Add README for seed 126
5802c2c verified

MNIST CNN Model (Seed 126)

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

Run Details

How to Use

import torch

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