reqst

dingguanglei · Oct 25, 2018 · da81526 · da81526
1 parent 0deb67d
commit da81526
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 17 deletions.
diff --git a/.gitignore b/.gitignore
@@ -17,7 +17,3 @@ Desktop.ini
 *.egg-info
 dist
 build
-
-# My configurations:
-db.ini
-deploy_key_rsa
diff --git a/docs/source/optimizer.rst b/docs/source/optimizer.rst
@@ -2,13 +2,13 @@
     :class: hidden-section
 
 jdit.optimizer
-==========
+==============
 
 .. automodule:: jdit
 .. currentmodule:: jdit
 
 Optimizer
------
+---------
 
 .. autoclass:: jdit.Optimizer
     :members:

diff --git a/jdit/model.py b/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
-from .pkg import *
+import torch, os
+from torch.nn import init, Conv2d, Linear, ConvTranspose2d, InstanceNorm2d, BatchNorm2d, DataParallel
+from torch import save, load
+
 class Model(object):
     """a model
 

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

diff --git a/jdit/pkg.py b/jdit/pkg.py