想问一下在使用多个GPU的时候,batch_size的问题..
xuanzebi opened this issue · comments
YP Han commented
Yuxian Meng commented
注意不要在其它地方设置环境变量如CUDA_VISIBLE_DEVICES,只用pytorch-lightning的n-gpus参数设置就没有问题
Code for ACL 2020 paper `A Unified MRC Framework for Named Entity Recognition`
xuanzebi opened this issue · comments
注意不要在其它地方设置环境变量如CUDA_VISIBLE_DEVICES,只用pytorch-lightning的n-gpus参数设置就没有问题