吴恩达卷积神经网络第一周作业PyTorch版本代码(gpu cpu通用)
吴恩达卷积神经网络,第一周作业PyTorch版本代码(gpu-cpu通用) 1.PyCharm上运行的PyTorch项目 2.基础的卷积神经网络搭建 3.加入了gpu加速所需的代码 4.含数据集+cnn_utils.py【对原版本做了简化】 5.含训练、模型保存、模型加载、单个图片预测代码 6.里面保存了个已经在gpu上训练好的模型,下载后也可以自行训练
文件列表
cnn.zip
(预估有个17文件)
cnn
CNN.py
1KB
Project1_use.py
780B
cnn_utils.py
1KB
.idea
misc.xml
304B
other.xml
239B
workspace.xml
10KB
cnn.iml
633B
inspectionProfiles
暂无评论