深度学习编程入门deep-learning-for-image-processing-master 关于图片分类test1_official_demo的学习
这个小文件夹有三个部分组成,分别有model,predict和train首先从train开始学习importtorchimporttorchvisionimporttorch.nnasnnfrommodelimportLeNetimporttorch.optimasoptimimporttorchvision.transformsastransformsdefmain():transform=tr