Tensorflow的启动比pytorch慢十倍

from timeit import default_timer as timer
import os

print('import pytorch')
start = timer()
import torch

end = timer()
print('Elapsed time: ' + str(end - start))

print('import tensorflow')
start = timer()
import tensorflow

end = timer()
print('Elapsed time: ' + str(end - start))

输出

import pytorch
Elapsed time: 0.46338812075555325
import tensorflow
Elapsed time: 4.396180961281061

回想起HanLP1.x时代为了加速模型加载连毫秒都要优化,现在我觉得非常失望。

刚给tf提了个issue,不知道官方有无改进的可能。

官方回复内部正在优化。比较影响调试体验,希望能早日用上。

1 Like

老师还是用pytorch重写把,哈哈

哈哈,如果HanLP是一个纯研究性的项目我会毫不犹豫地用PyTorch的。然而人在江湖,总有身不由己的地方。GitHub上也有类似的讨论,但还需要探索一段时间,甚至一个过渡期。