yolo3-pytorch icon indicating copy to clipboard operation
yolo3-pytorch copied to clipboard

train设置

Open chen-yuu opened this issue 3 years ago • 3 comments

1.为什么train的时候设置了with torch.no_grad()? 2.当目标检测只有一类且不需要类别信息的时候要怎么设置?

chen-yuu avatar Nov 04 '21 03:11 chen-yuu

1、因为图片不需要反传,反传的是网络的权值。 2、你可以认为目标只有一类且有类别信息

bubbliiiing avatar Nov 04 '21 09:11 bubbliiiing

还有请问train代码里 最后出现的两个if true指的是指的什么满足什么条件

chen-yuu avatar Nov 07 '21 13:11 chen-yuu

只是分为两个阶段

bubbliiiing avatar Nov 08 '21 16:11 bubbliiiing