Google colab是谷歌推出的一个免费的深度学习在线训练平台。google 的注册用户可以在上面跑基于TensorFlow、pytorch等架构的深度学习代码。其基本操作形式类似于jupyter。注册完google账号后(注册方法),即可以使用该功能。该平台虚拟出内存、CPU、GPU和TPU等供免费使用24小时。
默认情况下,运行的程序文件、数据集和模型文件存放在虚拟的硬盘中,一旦断开,任何数据都将无情的消失。因此,为了向模型提供训练和测试数据集、保存中间的checkpoint和最终的训练结果,需要利用google云盘来实现。一般来说,加载云盘的方式为在colab记事本中输入
from google.colab import drive
drive.mount('/content/drive')
执行后按要求输入driver的code会有 Mounted at /content/drive 的输出提示。此时在colab记事本左侧会出现如图所示的提示,证明挂载成功。
接着利用os.chdir命令可以实现对当前文件夹的切换,os.makedirs创建文件夹。具体来说,文件基本操作是这样的,效果如下图。
import os
os.chdir('/content/drive/My Drive/') #切换目录,这个是网盘的根目录
os.getcwd()#获取当前目录
if not os.path.exists('data'):
os.makedirs('data') #根目录下如果没有data文件夹,则创建data文件夹
os.chdir('data') #切换到data目录
os.chdir('..') #切换到上级目录