FB15-237数据集上的训练时间
Closed this issue · 3 comments
Infinempty commented
请问你们在fb15k-237数据集上训练模型的时间大概有多久呢，我运行了一下午没有一条日志信息，训练设备是3090
Infinempty commented
使用的命令是python -W ignore train.py --data_path=data/fb15k-237
yzhangee commented
yes, the running time on fb15k-237 per epoch is a bit long. if you have a larger GPU device, like A100, the batch size and testing batch size can be much larger, leading to faster running time.
Infinempty commented
ok，thank you for your reply