FastAI V2:learn.export（）中的tablerlearner export.pkl非常大

Question

我不确定这是否是有意的，但是来自learn.export()的export.pkl大约是471MB，这在某些应用程序中的部署中有点禁止

来自SaveModelCallback的模型本身只是131 KB，我只想使用Learner来应用相同的转换/处理（规范化、填充缺失、分类）

这有这么大的原因吗？我也确认了

learn.xb= （无）

learn.yb= （无）

Answer 1

原件：https://forums.fast.ai/t/tabularlearner-export-pkl-from-learn-export-is-very-large/81251/2

必须pip install wwf参见https://walkwithfastai.com/tab.export

from wwf.tab.export import *

1. 我们在学习者中手动保存模型 torch.save(learn.model, f'{model_dir}/2_{REF}_LEARNER_MODEL.pt')

2. 我们也导出表格对象 to.export(f'{model_dir}/3_{REF}_TABULAR_OBJECT.pkl')

3. 我们加载表格对象

to_new = load_pandas(f'{model_dir}/3_{REF}_TABULAR_OBJECT.pkl')
to_new = to_new.train.new(df[:20])
to_new.process()

1. 我们加载模型

model_2 = torch.load(f'{model_dir}/2_{REF}_LEARNER_MODEL.pt')
learn_new = TabularLearner(dls_new, model)

1. 我们做推理

row, clas, probs = learn_new.predict(df.iloc[0])
row.show()
probs

---

节省了大量资金：

• 型号：135 kb
• 表格对象：6KB

vs

• learn.export（）417 mb