move gravesnet.py

ShigekiKarita · Jul 29, 2015 · b3310a1 · b3310a1
1 parent 1037458
commit b3310a1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/test/test_gaussian_mixture_2d.py b/test/test_gaussian_mixture_2d.py
@@ -2,7 +2,6 @@
 
 import numpy
 from numpy.random import uniform, binomial
-
 import chainer
 from chainer import cuda
 from chainer import gradient_check
@@ -11,8 +10,7 @@
 
 from src.gaussian_mixture_2d import gaussian_mixture_2d
 from src.gaussian_mixture_2d_ref import gaussian_mixture_2d_ref
-from src import gravesnet
-
+import gravesnet
 
 if cuda.available:
     cuda.init()

diff --git a/train_prednet.py b/train_prednet.py
@@ -5,11 +5,10 @@
 import chainer
 import chainer.cuda
 import chainer.optimizers
-
 import six
 import numpy
 
-from src.gravesnet import GravesPredictionNet
+from gravesnet import GravesPredictionNet
 
 parser = argparse.ArgumentParser()
 parser.add_argument('--model', default='res/model', type=str,