Alexnet_pytorch_單GPU實現版本

純小白,純記錄 一、環境 ubuntu 18.04 CUDA 9.0 Cudnn 7.0 Opencv conda3 pycharm pytorch 二、簡介 使用Alexnet 網絡,識別貓狗圖片的分類。機子性能原因,只使用了22張圖片,epoch 只迭代了10次,只實現了訓練代碼,純學習pytorch,Alexnet用的。 三、網絡結構圖 四、代碼 代碼和數據:https://download
相關文章
相關標籤/搜索