Problem abput inplace_abn setup about hrnet-semantic-segmentation

[666shaw]

I though the author had given the answer in the .py file in models->bn_helpers.py, as follow:
import torch
import functools
  |  
  | if torch.version.startswith('0'):
  | from .sync_bn.inplace_abn.bn import InPlaceABNSync
  | BatchNorm2d = functools.partial(InPlaceABNSync, activation='none')
  | BatchNorm2d_class = InPlaceABNSync
  | relu_inplace = False
  | else:
  | BatchNorm2d_class = BatchNorm2d = torch.nn.SyncBatchNorm
  | relu_inplace = True