no gpu support for Matting mode?

[shartoo]

Thank you for your sharing.When i want to train matting mode,i found it can be trained in CPU mode but can never be trained on GPU mode with envirnoment as below

• cuda 9.0
• cudnn 7.3.1
• chainer (3.4 or 6.0rc)
• cupy (4.0rc or 6.0rc)

There are always sth related with cuda version or option error.Can this repo be trained on matting mode with GPU support?

[billzw]

Hav

Thank you for your sharing.When i want to train matting mode,i found it can be trained in CPU mode but can never be trained on GPU mode with envirnoment as below

• cuda 9.0
• cudnn 7.3.1
• chainer (3.4 or 6.0rc)
• cupy (4.0rc or 6.0rc)

There are always sth related with cuda version or option error.Can this repo be trained on matting mode with GPU support?

Have you resolved ?

[billzw]

def to_gpu(self, device=None): ''' Send layers to GPU except matting_link. ''' with chainer.cuda._get_device(device): super(chainer.Chain, self).to_gpu() d = self.dict for name in self._children: if name == 'matting_link': logger.info('Skip sending matting_link to GPU') continue d[name].to_gpu() return self matting_link

[shartoo]

Not yet,it seems that this repo is not well written ,there are many code bug refer here