pytorch把load的数据map到cpu上

发表于 2019-10-30 更新于 2020-05-07 阅读次数：
本文字数： 174 阅读时长 ≈ 1 分钟

1 2	checkpoint = torch.load("checkpoint.pth", map_location=torch.device('cpu')) model.load_state_dict(checkpoint["state_dict"])

第二句更具情况可有可无。