学习心得:
本次内容是通过task3训练好的模型来分别图像类型,大致操作与task2相似,本次我主要花费大量时间在cuda利用本地n显卡进行训练。也踩坑不少,总结了一些经验(本次实验源码由子豪同学开源)。
环境配置的踩坑:
主要是子豪同学示例代码中的torch库安装的torch为cpu版:
所以在torch库安装中一定要看好自己的cuda配置进行相应版本的安装,安装教程和踩坑教程都有很大的帮助。下图为配置正确的截图:
本次内容是通过task3训练好的模型来分别图像类型,大致操作与task2相似,本次我主要花费大量时间在cuda利用本地n显卡进行训练。也踩坑不少,总结了一些经验(本次实验源码由子豪同学开源)。
主要是子豪同学示例代码中的torch库安装的torch为cpu版:
所以在torch库安装中一定要看好自己的cuda配置进行相应版本的安装,安装教程和踩坑教程都有很大的帮助。下图为配置正确的截图: