dejanbatanjac:

to(0)


i found the VoxResNet architecture in github
this is the link where i found it  https://github.com/Ryo-Ito/brain_segmentation 7
but I made modifications according to my needs
and when I try vrn.to (‘cuda’) or vrn.to (0) I get this error
Traceback (most recent call last):
  File "train.py", line 169, in <module>
    main()
  File "train.py", line 94, in main
    optimizer.setup(vrn)
  File "/usr/local/lib/python3.6/dist-packages/chainer/optimizer.py", line 629, in setup
    super(GradientMethod, self).setup(link)
  File "/usr/local/lib/python3.6/dist-packages/chainer/optimizer.py", line 417, in setup
    raise TypeError('optimization target must be a link')
TypeError: optimization target must be a link