flxowens
/

mnist-classifier

Model card Files Files and versions Community

mnist-classifier / models

2 contributors

History: 5 commits

harry

feat: add prediction script and model file for MNIST digit classification

0dee387 2 months ago

mnist_model_lr0.001_bs128_ep10.pth
Detected Pickle imports (3)
- "torch.FloatStorage",
- "collections.OrderedDict",
- "torch._utils._rebuild_tensor_v2"
What is a pickle import?
4.8 MB
LFS

feat: enhance training loop with tqdm progress bar and configurable parameters 2 months ago
mnist_model_lr0.001_bs32_ep10.pth
Detected Pickle imports (3)
- "torch.FloatStorage",
- "collections.OrderedDict",
- "torch._utils._rebuild_tensor_v2"
What is a pickle import?
4.8 MB
LFS

feat: update training loop with learning rate scheduler and progress bar enhancements 2 months ago
mnist_model_lr0.001_bs64_ep10.pth
Detected Pickle imports (3)
- "torch.FloatStorage",
- "collections.OrderedDict",
- "torch._utils._rebuild_tensor_v2"
What is a pickle import?
4.8 MB
LFS

feat: update training loop with learning rate scheduler and progress bar enhancements 2 months ago
mnist_model_lr0.001_bs64_ep20.pth
Detected Pickle imports (3)
- "torch.FloatStorage",
- "collections.OrderedDict",
- "torch._utils._rebuild_tensor_v2"
What is a pickle import?
4.8 MB
LFS

feat: add prediction script and model file for MNIST digit classification 2 months ago