------------------------------------------------------------------------------

Copyright (c) Microsoft

Licensed under the MIT License.

Create by Bin Xiao (Bin.Xiao@microsoft.com)

Modified by Tianheng Cheng(tianhengcheng@gmail.com), Yang Zhao

------------------------------------------------------------------------------

from future import absolute_import from future import division from future import print_function

import os import logging

import torch import torch.nn as nn import torch.nn.functional as F

BatchNorm2d = nn.BatchNorm2d