1. 首页
  2. 人工智能
  3. 深度学习
  4. 吴恩达卷积神经网络第一周作业PyTorch版本代码(gpu cpu通用)

吴恩达卷积神经网络第一周作业PyTorch版本代码(gpu cpu通用)

上传者: 2020-07-16 20:09:07上传 ZIP文件 8.9MB 热度 64次
吴恩达卷积神经网络,第一周作业PyTorch版本代码(gpu-cpu通用) 1.PyCharm上运行的PyTorch项目 2.基础的卷积神经网络搭建 3.加入了gpu加速所需的代码 4.含数据集+cnn_utils.py【对原版本做了简化】 5.含训练、模型保存、模型加载、单个图片预测代码 6.里面保存了个已经在gpu上训练好的模型,下载后也可以自行训练
下载地址
用户评论