diff --git a/train.py b/train.py index 4611405e..30a07f3e 100644 --- a/train.py +++ b/train.py @@ -8,7 +8,6 @@ import test # Import test.py to get mAP after each epoch from models import * from utils.datasets import * from utils.utils import * -import torch.distributed as dist def train(