mod

hitokun-s · Apr 2, 2016 · 7d46b4e · 7d46b4e
1 parent 1f8f855
commit 7d46b4e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/imagenet/food_hack/predict.py b/imagenet/food_hack/predict.py
@@ -55,8 +55,8 @@ def mnist_predict(x, model):
 optimizer = optimizers.MomentumSGD(lr=0.01, momentum=0.9)
 optimizer.setup(model)
 
-serializers.load_npz(os.path.dirname(__file__) + 'model', model)
-serializers.load_npz(os.path.dirname(__file__) + 'state', optimizer)
+serializers.load_npz(os.path.join(os.path.dirname(__file__), 'model'), model)
+serializers.load_npz(os.path.join(os.path.dirname(__file__), 'state'), optimizer)
 
 
 mean_image = pickle.load(open("mean.npy", 'rb'))