Skip to content
#

cifar

Here are 65 public repositories matching this topic...

Comprehensive image classification for training multilayer perceptron (MLP), LeNet, LeNet5, conv2, conv4, conv6, VGG11, VGG13, VGG16, VGG19 with batch normalization, ResNet18, ResNet34, ResNet50, MobilNetV2 on MNIST, CIFAR10, CIFAR100, and ImageNet1K.

  • Updated Oct 5, 2021
  • Python

为了一劳永逸地解决 CIFAR 数据集模型训练问题,本文借鉴了多篇论文的模型训练代码,编写了基于 pytorch 的 CIFAR 数据集模型训练框架(在此我们简单的将该框架称为 CMTF)。在代码编写过程中,我发现 CMTF 可能对低性能的 GPU 设备更友好。CMTF 采用简单高效的训练配置,具有清晰的 log 风格,支持 VGG(VGG11、13、16、19及其带Batchnormal版本)和 ResNET(ResNet20、32、44、56、110)架构的训练。目前在CIFAR10上训练了 VGG16BN 和 ResNet20 模型上得到了checkpoint,获得了不低于一些学术论文的baseline精度。此外,CMTF 不需设置多卡并行计算,仅需简单操作即可添加新的模型结构

  • Updated Apr 16, 2023
  • Python

Improve this page

Add a description, image, and links to the cifar topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the cifar topic, visit your repo's landing page and select "manage topics."

Learn more