From 23a5cfb2d784aafa833f0abc908ce531e0b9bf72 Mon Sep 17 00:00:00 2001 From: hvy Date: Wed, 22 Nov 2017 17:22:17 +0000 Subject: [PATCH] Update to Chainer 3.0 --- lib/models/vgg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/models/vgg.py b/lib/models/vgg.py index 6f82aaf..2fc2e8d 100644 --- 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