mirrors/mmcv
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmcv.git synced 2025-06-03 21:54:52 +08:00
Code Issues Projects Releases Wiki Activity

support pytorch 1.0

Browse Source
This commit is contained in:
Kai Chen 2018-12-09 17:24:32 +08:00
parent a097c65fbf
commit 076cdd6c74
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmcv/image/io.py
View File

@ -54,7 +54,7 @@ def imfrombytes(content, flag='color'):
Returns: Returns:
ndarray: Loaded image array. 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 flag = imread_flags[flag] if is_str(flag) else flag
img = cv2.imdecode(img_np, flag) img = cv2.imdecode(img_np, flag)
return img return img

2
mmcv/parallel/distributed.py
View File

@ -33,7 +33,7 @@ class MMDistributedDataParallel(nn.Module):
self._dist_broadcast_coalesced(module_states, self._dist_broadcast_coalesced(module_states,
self.broadcast_bucket_size) self.broadcast_bucket_size)
if self.broadcast_buffers: 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: if len(buffers) > 0:
self._dist_broadcast_coalesced(buffers, self._dist_broadcast_coalesced(buffers,
self.broadcast_bucket_size) self.broadcast_bucket_size)

2
mmcv/runner/utils.py
View File

@ -13,7 +13,7 @@ def get_host_info():
def get_dist_info(): def get_dist_info():
if dist._initialized: if dist.is_initialized():
rank = dist.get_rank() rank = dist.get_rank()
world_size = dist.get_world_size() world_size = dist.get_world_size()
else: else:

2
mmcv/version.py
View File

@ -1 +1 @@
__version__ = '0.2.0' __version__ = '0.2.2'