From 0e1b91f74a1a84880bc38150e22b071f25265d20 Mon Sep 17 00:00:00 2001 From: liaoxingyu Date: Mon, 28 Dec 2020 14:35:02 +0800 Subject: [PATCH] fix checkpoint bug in ddp training Summary: change pytorch ddp to apex ddp in checkpoint --- fastreid/utils/checkpoint.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastreid/utils/checkpoint.py b/fastreid/utils/checkpoint.py index b01036c..7c32b1a 100644 --- a/fastreid/utils/checkpoint.py +++ b/fastreid/utils/checkpoint.py @@ -11,11 +11,11 @@ from typing import Optional, List, Dict, NamedTuple, Tuple, Iterable import numpy as np import torch import torch.nn as nn +from apex.parallel import DistributedDataParallel from termcolor import colored -from torch.nn.parallel import DataParallel, DistributedDataParallel +from torch.nn.parallel import DataParallel from fastreid.utils.file_io import PathManager -from fastreid.utils.events import get_event_storage class _IncompatibleKeys(