Alexnet_pytorch_单GPU实现版本

纯小白,纯记录 一、环境 ubuntu 18.04 CUDA 9.0 Cudnn 7.0 Opencv conda3 pycharm pytorch 二、简介 使用Alexnet 网络,识别猫狗图片的分类。机子性能原因,只使用了22张图片,epoch 只迭代了10次,只实现了训练代码,纯学习pytorch,Alexnet用的。 三、网络结构图 四、代码 代码和数据:https://download
相关文章
相关标签/搜索