2020年9月25日
RuntimeError: Trying to backward through the graph a second time, but the buffers have already been freed. Specify retain_graph=True when calling backward the first time.
摘要： 报错： 修改： model.module.optimizer_G.zero_grad() loss_G.backward() model.module.optimizer_G.step() 为： model.module.optimizer_G.zero_grad() loss_G.backward 阅读全文
posted @ 2020-09-25 15:47 皮卡皮卡妞 阅读(2996) 评论(0) 推荐(0)
vscode常用快捷键
摘要： 一、vs code 的常用快捷键 1、注释： a) 单行注释：[ctrl+k,ctrl+c] 或 ctrl+/ b) 取消单行注释：[ctrl+k,ctrl+u] (按下ctrl不放，再按k + u) c) 多行注释：[alt+shift+A] d) 多行注释：/** 2、移动行：alt+up/do 阅读全文
posted @ 2020-09-25 15:16 皮卡皮卡妞 阅读(510) 评论(1) 推荐(1)
pytorch中.pth文件转成.bin的二进制文件
摘要： model_dict = torch.load(save_path) fp = open('model_parameter.bin', 'wb') weight_count = 0 num=1 for k, v in model_dict.items(): print(k,num) num=num+ 阅读全文
posted @ 2020-09-25 15:13 皮卡皮卡妞 阅读(5364) 评论(0) 推荐(0)
AttributeError: module 'torch.jit' has no attribute 'get_trace_graph
摘要： pytorch+python 错误： 纠正： self.trace, _ = torch.jit.get_trace_graph(self.model, args=(self.x,)) 为： self.trace, _ = torch.jit._get_trace_graph(self.model, 阅读全文
posted @ 2020-09-25 15:09 皮卡皮卡妞 阅读(3196) 评论(0) 推荐(0)