请问下这个耗时大概多久

wenwenyu / PICK-pytorch

Code for the paper "PICK: Processing Key Information Extraction from Documents using Improved Graph Learning-Convolutional Networks" (ICPR 2020)

https://arxiv.org/abs/2004.07464

MIT License

559 stars 193 forks source link

请问下这个耗时大概多久 #11

Closed fxwfzsxyq closed 4 years ago

wenwenyu commented 4 years ago

@fxwfzsxyq 耗时是根据数据来定，在我们的实验中，文本框个数在100，文本最大长度50的设置下，8 卡V100， Batch Size 2*8，样本量在2100的情况下，训练阶段每个epoch是耗时35分钟。在测试阶段，CRF的Viterbi解码是比较耗时的步骤，其解码时间是和需要解码的文档长度成正比的。

wenwenyu commented 4 years ago

I am going to close this issue due to no response. Please feel free to reopen or create a new one if you have more questions.

y111x commented 3 years ago

@fxwfzsxyq 耗时是根据数据来定，在我们的实验中，文本框个数在100，文本最大长度50的设置下，8 卡V100， Batch Size 2*8，样本量在2100的情况下，训练阶段每个epoch是耗时35分钟。在测试阶段，CRF的Viterbi解码是比较耗时的步骤，其解码时间是和需要解码的文档长度成正比的。

您好，我发现训练时使用的GPU，验证的时候使用的CPU并且耗时较长，请问下有可能是验证的时候CRF解码的时候使用的cpu所以耗时较长吗？

y111x commented 3 years ago

我发现分布式运算时维特比解码速度很快，不到1秒：但是单GPU运算时，维特比解码会放到CPU上去运行，速度大概40秒 QQ截图20201111114941