boya34

Results 2 comments of boya34

您好,可以看看具体是怎么改的嘛。我也想改成多gpu,但目前还没有成功 > 谢谢答复,我改了trainer中的参数“ distributed_backend: dp”以及gpu数量,以及一些小的bug调整,能调用多gpu但是速度还没单卡快,不知为啥o(╥﹏╥)o

change `@cupy.util.memoize` to `@cupy.memoize` might be helpful