使用colab训练神经网络

流程:上传数据集到自己的google drive => 新建google colaboratory => 把colab和drive连接起来以访问drive上的数据集、代码文件等。

1.从google云盘新建colab的ipynb文件,重命名,并且修改为GPU。

2.colab和drive的连接:

from google.colab import drive
drive.mount('/content/drive/')

3.更改工作目录:

1 import os
2 path = "/content/drive/My Drive/Colab_Test/deep-image-prior/"
3 os.chdir(path) 
4 os.listdir(path)
5 !ls

4.从github上下载代码运行.py文件:

!git clone https://github.com/migu***/s***

5.运行程序

# 运行该项目的main.py文件
!python train.py

对于我的程序需要修改的地方:

1 !pip install tensorboardcolab
2 from torch.utils.tensorboard import SummaryWriter

 

posted @ 2021-10-03 12:13  sushamu  阅读(101)  评论(0编辑  收藏  举报