reqst

docs/source/conf.py

@@ -12,13 +12,12 @@
 # add these directories to sys.path here. If the directory is relative to the
 # documentation root, use os.path.abspath to make it absolute, like shown here.
 #
-# import os
-# import sys
-# sys.path.insert(0, os.path.abspath('../../'))
-# import jdit
+import os
+import sys
+sys.path.insert(0, os.path.abspath('../../'))
+import jdit
 
 # -- Project information -----------------------------------------------------
-
 project = 'jdit'
 copyright = '2018, dingguanglei'
 author = 'dingguanglei'
@@ -80,8 +79,10 @@
 # The theme to use for HTML and HTML Help pages.  See the documentation for
 # a list of builtin themes.
 #sphinx_rtd_theme,alabaster
+import sphinx_rtd_theme
 html_theme = 'sphinx_rtd_theme'
 
+html_theme_path = [sphinx_rtd_theme.get_html_theme_path()]
 # Theme options are theme-specific and customize the look and feel of a theme
 # further.  For a list of options available for each theme, see the
 # documentation.

docs/source/metric.rst

@@ -2,14 +2,14 @@
     :class: hidden-section
 
 jdit.Metric
-==========
+===========
 
 .. automodule:: jdit.metric
 .. currentmodule:: jdit.metric
 
 Metric
------
+------
 
-.. autoclass:: metric
+.. autoclass:: Metric
     :members:
 

docs/source/trainer.rst

@@ -8,13 +8,13 @@ jdit.trainer
 .. currentmodule:: jdit.trainer
 
 SupTrainer
------
+----------
 
 .. autoclass:: SupTrainer
     :members:
 
 GanTrainer
------
+----------
 
 .. autoclass:: GanTrainer
     :members:

jdit/model.py

@@ -1,8 +1,8 @@
 # coding=utf-8
-import torch, os
-from torch.nn import init, Conv2d, Linear, ConvTranspose2d, InstanceNorm2d, BatchNorm2d, DataParallel
-from torch import save, load
-
+# import torch, os
+# from torch.nn import init, Conv2d, Linear, ConvTranspose2d, InstanceNorm2d, BatchNorm2d, DataParallel
+# from torch import save, load
+from .pkg import *
 class Model(object):
     """a model
 

jdit/optimizer.py

@@ -1,7 +1,7 @@
 # coding=utf-8
 
-from torch.optim import Adam, RMSprop
-
+# from torch.optim import Adam, RMSprop
+from .pkg import *
 
 class Optimizer(object):
     """This is a wrapper of `optimizer` class in pytorch.

jdit/pkg.py

@@ -0,0 +1,5 @@
+from torch.optim import Adam, RMSprop
+
+import torch, os
+from torch.nn import init, Conv2d, Linear, ConvTranspose2d, InstanceNorm2d, BatchNorm2d, DataParallel
+from torch import save, load