Fix training crash with custom batch size

andy-yun / pytorch-0.4-yolov3

Yet Another Implimentation of Pytroch 0.4.1 and YoloV3 on python3

MIT License

279 stars 72 forks source link

Closed mickolka closed 6 years ago

mickolka commented 6 years ago

May (and, very probably, will) crash if image shape changes amid batch