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项目