josedolot commited on
Commit
59410c2
·
1 Parent(s): 382f1a2

Upload utils/sync_batchnorm/__init__.py

Browse files
Files changed (1) hide show
  1. utils/sync_batchnorm/__init__.py +13 -0
utils/sync_batchnorm/__init__.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # -*- coding: utf-8 -*-
2
+ # File : __init__.py
3
+ # Author : Jiayuan Mao
4
+ # Email : maojiayuan@gmail.com
5
+ # Date : 27/01/2018
6
+ #
7
+ # This file is part of Synchronized-BatchNorm-PyTorch.
8
+ # https://github.com/vacancy/Synchronized-BatchNorm-PyTorch
9
+ # Distributed under MIT License.
10
+
11
+ from .batchnorm import SynchronizedBatchNorm1d, SynchronizedBatchNorm2d, SynchronizedBatchNorm3d
12
+ from .batchnorm import patch_sync_batchnorm, convert_model
13
+ from .replicate import DataParallelWithCallback, patch_replication_callback