我有一个大小为 17GB 的 hdf5 文件(图像数据集),我需要将其上传并在 Google Colab 中使用,并在此数据集上训练我的模型。但是,当我使用以下代码从我的驱动器上传时,运行时间会在身份验证过程后自动断开连接。是因为文件大小还是其他问题?有什么解决方案可以克服这个问题吗?
代码片段如下:
from pydrive.auth import GoogleAuth
from pydrive.drive import GoogleDrive
from google.colab import auth
from oauth2client.client import GoogleCredentials
#Authenticate and create the PyDrive client.
auth.authenticate_user()
gauth = GoogleAuth()
gauth.credentials = GoogleCredentials.get_application_default()
drive = GoogleDrive(gauth)
# Get the file
downloaded = drive.CreateFile({'id': 'my_id'}) #replace the my_id with id of file you want to access
downloaded.GetContentFile('dataset.hdf5')