标签: tensorflow keras pytorch
是否有经过培训并仅与他人共享模型后保存模型的方法?像常规脚本一样?由于网络是浮点矩阵的集合,是否可以仅提取这些训练后的权重并在新数据上运行以进行预测,而不需要用户也安装这些框架?我是这些框架的新手,将根据需要进行任何澄清。
答案 0 :(得分:1)
PyTorch:如this post中所述,您可以将模型的参数另存为字典,或加载字典以设置模型的参数。 您也可以将PyTorch模型保存/加载为对象。 这两个过程都要求用户安装至少一个张量计算框架,例如进行有效的矩阵乘法。