From 076cdd6c74bbe5d0d16e1ef283b856c9487120c5 Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Sun, 9 Dec 2018 17:24:32 +0800
Subject: [PATCH] support pytorch 1.0

---
 mmcv/image/io.py             | 2 +-
 mmcv/parallel/distributed.py | 2 +-
 mmcv/runner/utils.py         | 2 +-
 mmcv/version.py              | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/mmcv/image/io.py b/mmcv/image/io.py
index 401283b78..0f19976c1 100644
--- a/mmcv/image/io.py
+++ b/mmcv/image/io.py
@@ -54,7 +54,7 @@ def imfrombytes(content, flag='color'):
     Returns:
         ndarray: Loaded image array.
     """
-    img_np = np.fromstring(content, np.uint8)
+    img_np = np.frombuffer(content, np.uint8)
     flag = imread_flags[flag] if is_str(flag) else flag
     img = cv2.imdecode(img_np, flag)
     return img
diff --git a/mmcv/parallel/distributed.py b/mmcv/parallel/distributed.py
index 1fec22598..e315c6155 100644
--- a/mmcv/parallel/distributed.py
+++ b/mmcv/parallel/distributed.py
@@ -33,7 +33,7 @@ class MMDistributedDataParallel(nn.Module):
             self._dist_broadcast_coalesced(module_states,
                                            self.broadcast_bucket_size)
         if self.broadcast_buffers:
-            buffers = [b.data for b in self.module._all_buffers()]
+            buffers = [b.data for b in self.module.buffers()]
             if len(buffers) > 0:
                 self._dist_broadcast_coalesced(buffers,
                                                self.broadcast_bucket_size)
diff --git a/mmcv/runner/utils.py b/mmcv/runner/utils.py
index 73e889ec2..0c0583da0 100644
--- a/mmcv/runner/utils.py
+++ b/mmcv/runner/utils.py
@@ -13,7 +13,7 @@ def get_host_info():
 
 
 def get_dist_info():
-    if dist._initialized:
+    if dist.is_initialized():
         rank = dist.get_rank()
         world_size = dist.get_world_size()
     else:
diff --git a/mmcv/version.py b/mmcv/version.py
index 7fd229a32..020ed73d7 100644
--- a/mmcv/version.py
+++ b/mmcv/version.py
@@ -1 +1 @@
-__version__ = '0.2.0'
+__version__ = '0.2.2'