07 2022 档案
摘要：有三种方式可以进行调试： 一、print() 利用print()输出自己想要查看的内容到控制台，但该方法每次运行都会完整运行程序，并且需要提前设定想要查看的内容，非常不方便。 二、pdb.set_trace()（推荐） import pdb ... pdb.set_trace() # 想要断点的位置
阅读全文
摘要：RuntimeError: Error(s) in loading state_dict for Sequential: 该错误通常与使用了nn.DataParallel进行训练有关 是指模型中的参数key中字符串与torch.load获取的key中字符串不匹配 因此，我们只需要修改torch.lo
阅读全文
摘要：一、动量（momentum） 可以给优化器加上一个动量，可以有效缓解局部最优问题。 原理上就是让优化过程从 W = W - lr * dW 变成 V = momentum * V - lr * dW W = W + V 使用示例： from torch import optim ... model
阅读全文
摘要：安装ffmpeg需要yasm，lame，nasm，libx264等等 一、下载所需文件的压缩包 此处给出压缩包的下载地址 1. yasm： http://yasm.tortall.net/Download.html 笔者下载的为yasm-1.3.0.tar.gz 2. lame： https://j
阅读全文
摘要：一、后台运行命令：nohup 示例： nohup python3 -u xxx.py >> xxx.txt 2>&1 & -u（可选，指不开启缓冲区，xxx.txt实时更新） 2>&1（2为错误输出，1为标准输出，整体意思是将错误输出重定向到标准输出中，然后输入到指定文件中） 末尾的&（指后台运行）
阅读全文
摘要：目前常见的有两种方法： 一、保存和加载整个模型：torch.save() + torch.load() 示例： torch.save(model, '/path/xxx.pth') model = torch.load('/path/xxx.pth') 二、仅保存和加载参数：torch.save()
阅读全文
浙公网安备 33010602011771号