Skip to content

Y
yolov4-pytorch

huigs / yolov4-pytorch
与 Fork 源项目一致

从无法访问的项目Fork

通知 3
Star 1
Fork 0
Y
yolov4-pytorch

前往新版Gitcode，体验更适合开发者的 AI 搜索 >>

未验证 提交 bfa5c39a 编写于 作者: B Bubbliiiing 提交者： GitHub
浏览文件
操作

Update yolo_training.py

上级 5b0019ab
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
nets/yolo_training.py
浏览文件 @ bfa5c39a
...@@ -220,7 +220,7 @@ class YOLOLoss(nn.Module): ...@@ -220,7 +220,7 @@ class YOLOLoss(nn.Module):
num_pos = torch.sum(mask) num_pos = torch.sum(mask)
num_pos = torch.max(num_pos, torch.ones_like(num_pos)) num_pos = torch.max(num_pos, torch.ones_like(num_pos))
else: else:
num_pos = bs num_pos = bs/3
return loss, num_pos return loss, num_pos
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或