Closed piseabhijeet closed 3 years ago
Hi @weixmath
How do i enable multi-GPU training? I always end up with CUDA ALLOCATION error.
I trained this model on 12G 1080Ti and i think the training is fast enough. Maybe you can modify it by yourself.
@piseabhijeet @weixmath Hi, the multi-GPU implementation is available here !
Hi @weixmath
How do i enable multi-GPU training? I always end up with CUDA ALLOCATION error.