# 直接加载模型
model.load_state_dict(torch.load('./data/my_model.pkl'))

#GPU训练的模型加载到CPU上：
model.load_state_dict(torch.load('./data/my_model.pkl', map_location=lambda storage, loc: storage))

#加载到GPU1上：
model.load_state_dict(torch.load('./data/my_model.pkl', map_location=lambda storage, loc: storage.cuda(1)))

#从GPU1 移动到 GPU0：
model.load_state_dict(torch.load('./data/my_model.pkl', map_location={'cuda:1':'cuda:0'}))

 

# 深度学习
Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

cover

Coze开发AI翻译应用

avatar 2048 AI社区
cover

LangGraph 入门

avatar 2048 AI社区

演练：使用VB开发多智能体协作的荣格八维分析器 在大

荣格八维理论是心理学家卡尔·荣格提出的认知功能理论，后发展为多个分支，其中人气较高的是 MBTI。该理论认为人的认知功能可以分为八种，在不同的位置中担任不同的原型。这些功能随着人的成长而发展，并且具有先天性。通过这个项目，我展示了如何使用 Visual Basic .NET 构建一个复杂的多智能体协作系统。强制工作流：通过硬编码待办事项列表确保分析按预期顺序进行，部分子智能体会强制调用指定的工具具

avatar 2048 AI社区