Transformer模型训练好以后如何固化？

由cqs16800创建，最终由cqs16800更新于2023-06-01 02:13 被浏览 38 用户

如题，训练一次Transformer模型很不容易，训练完成以后如何固化呢？用固化深度模型的方法没办法存储。

提示如下：

ValueError Traceback (most recent call last) <ipython-input-17-6cbf37e6754a> in <module> 1 print(m4.data) 2 ds = m4.data ----> 3 pd.DataFrame([DataSource(ds.id).read()]).to_pickle('/home/bigquant/work/Transformer最简版.csv')

ValueError: Must pass 2-d input. shape=(1, 669547, 1)

这个提示貌似是说要存为张量形式，不知如何解决，敬请攻城狮和各位大神指教