- 博客(2)
- 收藏
- 关注
原创 保存与加载PyTorch训练的模型和超参数--基于kaggle树叶分类
如今,在深度学习训练所需要的时间以数小时起步的现状下,如果不保存训练好的最佳权重参数,或者一旦突然中断训练,会耗费很多时间成本。本篇将介绍如何使用pytorch保存你已训练得到的最佳参数,并且在下次继续训练时能以此参数为起点。
2024-05-31 15:26:02 364
原创 pytorch+anaconda+vscode深度学习环境配置(GPU)
pytorch+anaconda+vscode深度学习环境配置(GPU)默认已搭好anaconda+vscode的Python环境,可参考https://blog.csdn.net/vinkim/article/details/81546333因为这里踩了很多坑,最后发现可能是安装顺序出了问题,所以务必按照以下顺序安装先下载CUDA,cuDNNhttps://developer.nvidia.com/cuda-toolkithttps://developer.nvidia.com/cudnn
2021-08-08 12:51:29 3908
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人