登录 注册
当前位置:主页 > 资源下载 > 50 > 吴恩达卷积神经网络第一周作业的PyTorch版本代码,适用于GPU和CPU

吴恩达卷积神经网络第一周作业的PyTorch版本代码,适用于GPU和CPU

  • 更新:2024-07-30 09:19:44
  • 大小:8.9MB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:深度学习 - 人工智能
  • 格式:ZIP

资源介绍

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