cnn_transfer / seed_103 /README.md
abir-hr196's picture
Add README for seed 103
02c7350 verified

MNIST CNN Model (Seed 103)

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

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: 103

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_103/pytorch_model.bin"))
# model.eval()