Python 将Pytorch model.state_dict（）保存到s3

我试图将一个经过训练的Pytorch模型保存到S3。但是，

torch.save（model.state\u dict（），file\u name）

似乎只支持本地文件。如何将状态dict保存到S3文件中

我使用的是Torch

0.4.0

正如所讨论的，Pytorch没有定制的API来完成这项工作。但是，您可以使用boto3或Petastorm库来解决此问题

下面是一个直接写入S3对象的具体示例：

import boto3

# Convert your existing model to JSON
saved_model = model.to_json()

# Write JSON object to S3 as "model.json"
client = boto3.client('s3')
client.put_object(Body=saved_model,
                  Bucket='BUCKET_NAME',
                  Key='model.json')