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

refactor apex import

pull/389/head
liaoxingyu 2021-01-23 15:35:48 +08:00
parent a53fd17874
commit ef6ebf451b
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
fastreid/utils/checkpoint.py
View File

@ -11,12 +11,17 @@ from typing import Optional, List, Dict, NamedTuple, Tuple, Iterable
import numpy as np import numpy as np
import torch import torch
import torch.nn as nn import torch.nn as nn
from apex.parallel import DistributedDataParallel
from termcolor import colored from termcolor import colored
from torch.nn.parallel import DataParallel from torch.nn.parallel import DataParallel
from fastreid.utils.file_io import PathManager from fastreid.utils.file_io import PathManager
try:
from apex.parallel import DistributedDataParallel
except ImportError:
raise ImportError("Please install apex from https://www.github.com/nvidia/apex to run this example if you want to"
"train with DDP")
class _IncompatibleKeys( class _IncompatibleKeys(
NamedTuple( NamedTuple(