pytorch_block_sparse icon indicating copy to clipboard operation
pytorch_block_sparse copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

does this package allow multi-gpu training and distributed training?

Open DanqingZ opened this issue 4 years ago • 1 comments

If so, could you provide an example? Thanks.

DanqingZ avatar Nov 11 '20 07:11 DanqingZ

Hi, I tried it with pytorch lightning using DDP and it works as expected

pclucas14 avatar Jun 22 '22 13:06 pclucas14