Python - 将字节 object 转换为张量 object 用于 torch.load()

Question

我已经从我的谷歌驱动器下载了一个tensor (.pt) object 到我的 Python 工作区，它目前是一个bytes object。它需要是一个tensor object (X) 才能在代码torch.load(X)中使用：

从我的 Google Drive 下载 tensorflow object 的代码是：

file_id2 = 'XXXXXXXXXXXXXXXXXXXXXXXXX'
request = drive_service.files().get_media(fileId=file_id2)
gh = io.BytesIO()

downloader = MediaIoBaseDownload(gh, request)
done = False
while done is False:
        status, done = downloader.next_chunk()
        print ("Download %d%%." % int(status.progress() * 100))
X = gh.getvalue()

我试图找到一种将字节 object (X) 转换为张量 object 的方法，但我没有找到任何有用的东西

有没有办法做到这一点？ 或从谷歌下载张量 object 作为.pt文件而不是bytes的方法。

Answer 1

我设法解决了它，所以将它留在这里以防将来有人需要它。

tensorObject = torch.load(io.BytesIO(X))

我必须将 object 作为 BytesIO object 传递（我认为它已经是），现在它可以工作了:)