PyTorch-YOLOv3
PyTorch-YOLOv3 copied to clipboard
Add support to multi-GPU training
-
[ ] add support to multi-GPU training by wrapping the model with
DtataParallel -
[ ] add function to automatically save the best checkpoint