toandaominh1997 / EfficientDet.Pytorch

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch
MIT License
1.44k stars 305 forks source link

num_epoch #51

Closed XiangjiBU closed 4 years ago

XiangjiBU commented 4 years ago

hi, thanks for sharing! In train.py, paramter "num_epoch" defaults 500. I thought the total iteration num is 500*1000, but it seem not. Could you explain this param ?

toandaominh1997 commented 4 years ago

@NoahBU Because I have saved weights for each epoch, I put a lot of epoch so my server can work during my free time. I can choose best weights based on the loss and accuracy of the validation set in each epoch:v