pytorch1.8

classification

所需环境

• Anaconda3（建议使用）
• python3.6 / 3.7
• pycharm (IDE)
• pytorch 1.8.1 (pip package)
• torchvision 0.7.0 (pip package)

集成了经典的分类模型

• Densennet
• EfficientNet v1/V2
• GooleNet
• Mobilenet V2/v3
• RegNet
• Resnet系列
• VGG系列
• ShuffleNet

默认开启tensorboard

• 端口6006
• runs文件夹为tensorboard项目