Programming Language:
Note
This is a PyTorch
version of TensorAI
.
Please install the requirements before using,
including torch==2.6.0
and torchvision==2.6.0
Model | Libs |
---|---|
TensorAI | TensorFlow and pillow |
PyTorchMNIST | PyTorch and pillow |
- Inference
- Download & install
Python-3.13
from Python.org - Download
PyTorchMNIST
: main.zip - Install requirements:
pip3 install -r requirements.txt
- Run the program (
run.py
)
- Download & install
- Train
python train.py
TEST Accuracy | TEST Loss |
---|---|
0.9915 (99.15%) | 0.0012 |