search

bubbliiiing / yolov4-tiny-pytorch

这是一个YoloV4-tiny-pytorch的源码，可以用于训练自己的模型。
MIT License
787 stars 184 forks source link

train.py脚本中的fit_one_epoch入参是不是有问题？ #38

Closed happy2019-k closed 3 years ago

happy2019-k commented 3 years ago

def fit_one_epoch(net,yolo_losses,epoch,epoch_size,epoch_size_val,gen,genval,Epoch,cuda):其中yolo_losses参数在后边并没有使用，而使用的是yolo_loss，这个地方是不是有问题？

happy2019-k commented 3 years ago

@bubbliiiing

bubbliiiing commented 3 years ago

忘了改变量名，不影响训练的。