cnn_transfer / seed_136 /README.md
abir-hr196's picture
Add README for seed 136
820ae32 verified

MNIST CNN Model (Seed 136)

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

Run Details

Model Performance

  • Best Validation Accuracy: 99.05%

Configuration

  • Architecture: CNN
  • Dataset: MNIST
  • Epochs: 10
  • Batch Size: 128
  • Learning Rate: 0.0003
  • Random Seed Used: 136

How to Use

import torch
# from mnist_transformer import MNISTCNN # Assuming MNISTCNN class definition

# model = MNISTCNN()
# model.load_state_dict(torch.hub.load_state_dict_from_url("https://huggingface.co/abir-hr196/cnn_transfer/resolve/main/seed_136/pytorch_model.bin"))
# model.eval()