Closed gouhaogou closed 3 years ago
请问您解决了么,我也遇到了相同的问题,应该是GPU的问题,我的设备只有一块GPU但是代码是4块并行运算的,但我不知到该怎么修改,请问您解决了吗?谢谢!
@gouhaogou @Dylan-SUFE 将 --gpus
参数进行修改即可。具体可参考pytorch-lightning的documentation
@YuxianMeng 谢谢,已解决
@YuxianMeng 谢谢,已解决
您好请问您是怎么解决的?修改哪里的gpu参数,能详细说说么?谢谢您了!
@YuxianMeng 谢谢,已解决
您好请问您是怎么解决的?修改哪里的gpu参数,能详细说说么?谢谢
你好,你也在做这个吗,可否留个QQ或其他联系方式,我们交流一下
@YuxianMeng 谢谢,已解决
您好请问您是怎么解决的?修改哪里的gpu参数,能详细说说么?谢谢
你好,你也在做这个吗,可否留个QQ或其他联系方式,我们交流一下
可以呀,qq:1627128918,最近在复现这篇论文。
环境 PyTorch Version : 1.7.1 OS : Windows 10 Python version: 3.6.12 CUDA/cuDNN version: 10.2 GPU models and configuration: 1050
以下 $ sh zh_msra.sh Some weights of the model checkpoint at D:/Git/mnt/mrc/chinese_roberta_wwm_large_ext_pytorch were not used when initializing BertQueryNER: ['cls.predictions.bias', 'cls.predictions.transform.dense.weight', 'cls.predictions.transform.dense.bias', 'cls.predictions.transform.LayerNorm.weight', 'cls.predictions.transform.LayerNorm.bias', 'cls.predictions.decoder.weight', 'cls.seq_relationship.weight', 'cls.seq_relationship.bias']