diff --git a/new_eval_all.py b/new_eval_all.py index eb69611d1a702b7f2dbd2ae93f2622dbb9a47ca9..d4c4cc1277e01ba738abb895437d10261d6f25bf 100644 --- a/new_eval_all.py +++ b/new_eval_all.py @@ -8,6 +8,7 @@ import tempfile import shutil import mmcv import torch + import torch.distributed as dist from mmcv.parallel import MMDataParallel, MMDistributedDataParallel from mmcv.runner import get_dist_info, load_checkpoint