2021年12月13日
优化器导入参数将Tensor结构转为cuda类型
摘要： optimizer.load_state_dict(checkpoint['optimizer']) for state in optimizer.state.values(): for k, v in state.items(): if isinstance(v, torch.Tensor): s 阅读全文
posted @ 2021-12-13 14:11 祥瑞哈哈哈 阅读(450) 评论(0) 推荐(0)