| | | |

2024年9月3日

摘要： PyTorch中有modelzoo还有一些训练好的权重参数可以直接使用，在迁移学习中就很方便 torchvision.models - PyTorch中文文档 (pytorch-cn.readthedocs.io) 看看上面中文文档就知道怎么调用模型了，要使用 torchvision.models 阅读全文

posted @ 2024-09-03 13:22 千主阅读(34) 评论(0) 推荐(0)

PyTorch使用GPU进行训练

摘要：要确保有GPU对应的CUDA版本下载好直接按照下图模块测测试调用GPU就好了这是测试GPU是否能使用。如果显示了可以使用后续使用操作如下：将实例化的模型传入GPU中：直接看一个训练的示例代码：要在GPU中处理使用都要传入GPU中才能用。阅读全文

posted @ 2024-09-03 11:59 千主阅读(78) 评论(0) 推荐(0)