search

WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.98k stars 524 forks source link

为何在做test的时候没有得到全部的结果，而是只有前三个batch的结果？ #101

Open JumeLin opened 2 years ago

JumeLin commented 2 years ago

我的测试集有151张，batchsize设置为4，但是在做测试的时候只保存了前三个batch的测试结果，可以给个解释吗？

WongKinYiu commented 2 years ago

https://github.com/WongKinYiu/yolor/blob/main/test.py#L222

JumeLin commented 2 years ago

https://github.com/WongKinYiu/yolor/blob/main/test.py#L222

谢谢，我找到了