search

yao8839836 / kg-bert

KG-BERT: BERT for Knowledge Graph Completion
Apache License 2.0
679 stars 141 forks source link

gpu的使用 #29

Open MiaoHu-Pro opened 2 years ago

MiaoHu-Pro commented 2 years ago

代码中，提供了两种多卡并行方式： DistributedDataParallel 和 DataParallel，

请问：1. 您在使用这两种方式进行训练时，运行时间差别大吗？

  1. 我使用 v-100 运行 link prediction 任务时，花了近1周时间还没结果，您是否有其他方式进行加速？

谢谢。