Tensorflow的启动比pytorch慢十倍

[hankcs]

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时代为了加速模型加载连毫秒都要优化，现在我觉得非常失望。

[hankcs]

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

import tensorflow is very slow

[hankcs]

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

[nixl]

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

[hankcs]

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