在serialization.py文件中，修改了load函数，现在它使用torch.device('cpu')作为默认的map_location，并且指定pickle_module为pickle，同时允许通过pickle_load_args传递其他加载参数。
def load(f, map_location=torch.device('cpu'), pickle_module=pickle, **pickle_load_args):
serialization.py文件更改一下load的方式
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
def load(f, map_location=torch.device('cpu'), pickle_module=pickle, **pickle_load_args):
serialization.py文件更改一下load的方式
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
