2022年2月23日
查看模型所在的cuda
摘要： 使用 model.device 阅读全文
posted @ 2022-02-23 15:17 呦呦南山 阅读(100) 评论(0) 推荐(0) 编辑
设置数据保存路径
摘要： print(f"./feature_model/save_feature_state_aug_{num_item[i]}_{k}_bcq")其中 num_item[i]，k是变量 阅读全文
posted @ 2022-02-23 15:00 呦呦南山 阅读(41) 评论(0) 推荐(0) 编辑
当数据和要加载的模型不在同一个cuda上时的处理方法
摘要： 问题： c = F.relu(policy_model.Q.c1(exp_list[i][k].unsqueeze(0)))error: *** RuntimeError: Expected all tensors to be on the same device, but found at lea 阅读全文
posted @ 2022-02-23 14:52 呦呦南山 阅读(3135) 评论(0) 推荐(0) 编辑