add distributed API for model, datasets, trainer.

dingguanglei · Jun 1, 2019 · 4c273b9 · 4c273b9
1 parent 4304b4d
commit 4c273b9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/jdit/model.py b/jdit/model.py
@@ -320,7 +320,7 @@ def convert_to_distributed(self, device_ids=None,
         >>> net.model = torch.nn.DistributedDataParallel(net.model, pg)
 
         """
-        assert isinstance(self.model, DataParallel), "please only use one gpu for one task"
+        # assert isinstance(self.model, DataParallel), "please only use one gpu for one task"
         self.model = DistributedDataParallel(self.model, device_ids,
                                              output_device, dim, broadcast_buffers,
                                              process_group, bucket_cap_mb,

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="jdit",  # pypi中的名称，pip或者easy_install安装时使用的名称，或生成egg文件的名称
-    version="0.0.14",
+    version="0.0.15",
     author="Guanglei Ding",
     author_email="dingguanglei.bupt@qq.com",
     maintainer='Guanglei Ding',