fix docs

dingguanglei · Aug 9, 2019 · f7c107c · f7c107c
1 parent 7b240c2
commit f7c107c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/jdit/assessment/fid.py b/jdit/assessment/fid.py
@@ -395,7 +395,7 @@ def FID_score(source, target, sample_prop=1.0, gpu_ids=(), dim=2048, batchsize=1
 
     """
     assert sample_prop <= 1 and sample_prop > 0, "sample_prop must between 0 and 1, but %s got" % sample_prop
-    model = _InceptionV3([_InceptionV3.BLOCK_INDEX_BY_DIM[dim]])
+    model = InceptionV3([InceptionV3.BLOCK_INDEX_BY_DIM[dim]])
     if isinstance(source, Tensor) and isinstance(target, Tensor):
         # source[?,C,H,W] target[?,C,H,W]
         s_length = len(source)

diff --git a/setup.py b/setup.py
@@ -15,7 +15,8 @@
     keywords="pytorch research framework",
     platforms=["all"],
     url="https://github.com/dingguanglei/jdit",
-    packages=['jdit','jdit/trainer',
+    packages=['jdit',
+              'jdit/trainer',
               'jdit/trainer/gan',
               'jdit/trainer/single',
               'jdit/trainer/instances',