Update to Chainer 3.0

hvy · Nov 22, 2017 · 23a5cfb · 23a5cfb
1 parent a59a905
commit 23a5cfb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/models/vgg.py b/lib/models/vgg.py
@@ -72,7 +72,7 @@ def feature_map_activations(self, x):
 
             # Disable cuDNN, else pooling indices will not be stored
             with chainer.using_config('use_cudnn', 'never'):
-                h = mp(h)
+                h = mp.apply((h,))[0]
             hs.append(h)
 
         return hs, pre_pooling_sizes