Skip to content

Haozhe-py/PyTorchMNIST

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorchMNIST

Programming Language:

Python

Note

This is a PyTorch version of TensorAI. Please install the requirements before using, including torch==2.6.0 and torchvision==2.6.0

Contents

  1. Versions
  2. Usage
  3. Accuracy&Loss

1. Versions

Model Libs
TensorAI TensorFlow and pillow
PyTorchMNIST PyTorch and pillow

2. Usage

  • Inference
    1. Download & install Python-3.13 from Python.org
    2. Download PyTorchMNIST : main.zip
    3. Install requirements: pip3 install -r requirements.txt
    4. Run the program (run.py)
  • Train
    • python train.py

3. Accuracy and Loss

TEST Accuracy TEST Loss
0.9915 (99.15%) 0.0012

Releases

No releases published

Packages

No packages published

Languages