- 使用pytorch框架 - 先保存为.pkl文件,然后加载模型预测 - 同批数据,模型预测耗时相差近一倍

- 使用pytorch框架。 - 先保存为.pkl文件,然后加载模型预测。 - 同批数据,模型预测耗时相差近一倍

bilstm_model = load_model(BiLSTM_MODEL_PATH)
#bilstm_model.model.bilstm.flatten_parameters()  # remove warning
start_time = time.time()
lstm_pred, target_tag_list = bilstm_model.test(test_word_lists, test_tag_lists, bilstm_word2id, bilstm_tag2id)

end_time = time.time()
print ("Cost time: ",(end_time - start_time) )

相关文章

贪吃蛇
Python程序,可以根据要求自动生成密码字典,并拆分成多个5M...
难以描述
simplified.py 精简字体
贪吃蛇
json python中的转储函数import json   # ...