UER-py
UER-py copied to clipboard
gate_cnn 模型GPU多卡训练报错
python环境:

GPU cuda配置

多卡跑bert模型预训练是可以的
一直不断的重复下面的信息:
Waiting in store based barrier to initialize process group for rank: 5, key: store_based_barrier_key:1 (world_size=4, worker_count=19, timeout=0:30:00)
GPU卡是:GeForce RTX 3090,BERT可以,gate_cnn为啥不行呢