From 0f6bd1d6cda40f9443a612d288bec58756184d53 Mon Sep 17 00:00:00 2001 From: hvy Date: Fri, 26 May 2017 03:30:28 +0000 Subject: [PATCH] Remove cupy dependency --- lib/models/vgg.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/lib/models/vgg.py b/lib/models/vgg.py index 708f5f2..14eafc9 100644 --- a/lib/models/vgg.py +++ b/lib/models/vgg.py @@ -1,8 +1,8 @@ -import cupy import chainer +from chainer import cuda +from chainer import Variable import chainer.links as L import chainer.functions as F -from chainer import Variable class VGG(chainer.Chain): @@ -127,8 +127,8 @@ def check_add_deconv_layers(self, nobias=True): for conv in conv_block: out_channels, in_channels, kh, kw = conv.W.data.shape - if isinstance(conv.W.data, cupy.ndarray): - initialW = cupy.asnumpy(conv.W.data) + if isinstance(conv.W.data, cuda.ndarray): + initialW = cuda.cupy.asnumpy(conv.W.data) else: initialW = conv.W.data @@ -138,7 +138,7 @@ def check_add_deconv_layers(self, nobias=True): initialW=initialW, nobias=nobias) - if isinstance(conv.W.data, cupy.ndarray): + if isinstance(conv.W.data, cuda.ndarray): deconv.to_gpu() self.add_link('de{}'.format(conv.name), deconv)