search

bubbliiiing / yolov7-pytorch

这是一个yolov7的库，可以用于训练自己的数据集。
GNU General Public License v3.0
844 stars 149 forks source link

reload dataset each epoch #80

Open trand2k opened 1 year ago

trand2k commented 1 year ago

hi, guy why line 537 in train.py file need re-load dataset per epoch? It already load in line 484. any trick for this?

bubbliiiing commented 1 year ago

Once only, because the batch_size changes when we unfreeze the backbone