chainer
OK なコード model = Regression(Model()) if gpu >= 0: xp = cuda.cupy cupy.cuda.Device(gpu).use() model.to_gpu() else: xp = np opt = optimizers.Adam() # モデルを gpu に移したあとで opt.setup(model) # optimizer と紐付けるとOK NGなコード model…
OK なコード model = Regression(Model()) if gpu >= 0: xp = cuda.cupy cupy.cuda.Device(gpu).use() model.to_gpu() else: xp = np opt = optimizers.Adam() # モデルを gpu に移したあとで opt.setup(model) # optimizer と紐付けるとOK NGなコード model…