【2】模型训练准备
星云 2024/7/5 图像分类
【2】模型训练准备
# 前言
构建完图像分类数据集后,就要开始训练我们的模型了,深度学习模型训练需要大量计算资源,也就是GPU。 可以在本地机器上使用GPU(如果有的话),或者在云服务上租用GPU资源。此外,还有专门为深度学习训练提供的服务,如Google的Colab和Kaggle。
本文将简单介绍一下如何使用kaggle上的GPU资源。
# 一、kaggle账户创建
注册完kaggle账号后,为了使用GPU资源,还要进行手机号验证
推荐文章
https://blog.csdn.net/jiabiao1602/article/details/139370991 (opens new window)
https://blog.csdn.net/weixin_51288849/article/details/130164188 (opens new window)
# 二、创建kaggle项目
由于考虑到kaggle云平台GPU使用时间的限制,我创建了一个kaggle项目让团队成员加入进来,登录他们的账号进行训练。
参考文章: 如何在kaggle中发布一个项目以便管理 (opens new window)
# 三、使用
将图像数据上传到kaggle项目中
相关notebook代码,请参见: kaggle_train.ipynb
# 总结
准备好了模型训练的资源,将数据上传到了kaggle平台,编写了相关代码。
2024/6/12
共赏金尊沈绿蚁,莫辞醉,此花不与群花比