mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix checkpoint bug in ddp training 

Summary: change pytorch ddp to apex ddp in checkpoint
pull/380/head
liaoxingyu 2020-12-28 14:35:02 +08:00
parent f56ca8345e
commit 0e1b91f74a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
fastreid/utils/checkpoint.py
View File

@ -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(