Training Stuck on epoch XXXXXXX start - Githubissues

longcw / yolo2-pytorch

YOLOv2 in PyTorch

1.54k stars 421 forks source link

Training Stuck on epoch XXXXXXX start #33

Closed lwoiceshyn closed 6 years ago

lwoiceshyn commented 6 years ago

Hi, This might just be because I'm using Python 3.6, but when I try and run train.py I just get stuck in an endless amount of: epoch 1238472147 start....

Is there any way to fix this with imdb.py or would I have to write my own version to load the dataset?

lwoiceshyn commented 6 years ago

For anyone else with this issue, simply change the imap method to map, as map in Python 3 is equivalent.