kaggle
进行模型训练的时候需要用到GPU进行运算,此时便需要较好的显卡。在大模型·面前1050Ti是不够用的。不过还好有kaggle——主要是为开发商和数据科学家提供举办机器学习竞赛、托管数据库、编写和分享代码的平台。它会免费提供云服务器。
kaggle是外国网站,以前国内可以访问只是不能上传数据。现在可通过网址访问。
网址为https://www.kaggle.com/
01
打开后右上角Sign in登陆,可以使用多种方式创建账号,按步骤创建即可。
进入后
02
选择notebook,new一个
03
根据爱好选择一种写代码方式,有需要可以点开设置是否打开cpu,gpu等,稍后也可以使用,然后create
导入数据集
aggle不像是使用你的电脑,它没法直接读取你的数据,所以使用的数据集需要一一上传。右上角有一个Add data添加数据集,当然也可以在主页添加。添加的数据集可以kaggle自带的,也可以上传自己的,自己的就右上角upload
下面使用一个四种花分类来演示
01
上传数据要压缩文件,再起个名字,然后create。
02
完成后input中就有了
03
然后在代码中路径data_dir=’…/input/flower/图片’ 要这么表示,在input目录下。我们的目的是得到训练完的权重或者模型,那么最后使用model.save(‘4flowers.h5’)经过训练后即可保存到output中
打开GPU或TPU
在settings中打开GPU或者TPU,这里使用的是GPU,如果使用TPU,使用前需要配置,配置方法百度即可。
保存并获取
为了得到训练好的模型,我们已经将他保存到output中了,直接点击下载标志即可下载到本地。
但有时候点击没反应,可能是因为卡顿或者模型特别大,这里就需要点击右上角的Save version, 并选取第二个详细保存
save后可以在notebook首页查看保存下来的完整信息,也可以用于分享和展示。
GPU等不是无限使用的,不过kaggle每周都给30h的gpu和tpu使用时间。
想看更多精彩内容
请关注我们的实验室公众号哟
扫码关注啦!