desti-nation / CatVSDog

猫狗大战 二分类问题

猫狗大战阉割版代码

环境说明

torchvision==0.2.1
torch==0.4.1
tensorboard_logger==0.1.0
tqdm==4.28.1
Pillow==5.3.0

文件说明

main.py 运行时的主函数

config.py 参数设置

model.py 模型

data.py Pytorch的数据读入函数

utils.py 工具集合