yjyjjyjy1231的博客

文章目录统计网络参数量统计网络参数量# 所有参数量p_all = sum([torch.numel(param) for param in model.parameters()])# 需进行梯度更新的参数量p_require_grad = sum([torch.numel(param) for param in model.parameters() if param.requires_g...

最近在调试yolov3的pytorch版本的程序，过程记录在yolov3-spp结构详解、源码解析及踩坑指南（pytorch）今天碰到了这样一个问题：在训练过程中，会突然停住（如下图），也不报错，但也不继续训练，就只是停住，仿佛时间静止了。如果强行点个暂停，发现程序是一直在tqdm的Tmoniter中的run函数中，一直在执行：if self.was_killed.is_set(): c...

文章目录bug1: 'too many indices for tensor of dimension 1'bug2: 多GPU训练 ''YOLOLayer' object has no attribute 'ng''bug1: ‘too many indices for tensor of dimension 1’bug描述：问题出在dataset.py文件中，collate_fn函数中有...