Skip to content

undersnow/hhh

Repository files navigation

PyTorch实现MNIST手写数字识别

这是学习计算机视觉时的练手的一个小项目,非常简单,主要是本人用来入门和熟悉PyTorch框架的

目前已经完成了简单的模型的构建,训练及测试,和将二进制存储的数据集转换为图片的小工具;计划日后添加一些可视化的功能

以后也有实践诸如 AlexNetVGGGooLeNetResNet 等视觉方向常用网络模型的打算

使用的环境:

  • Python 3.8.5(Anaconda)
  • PyTorch 1.7.1
  • CUDA 11.1

测试平台:

  • Windows 10(RTX3070)
  • Ubuntu 20.04(RTX3070)
  • Mac Os Big Sur(CPU)

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages