【发布时间】:2018-03-14 13:26:28
【问题描述】:
我已经通过 coremltools 将带有 LSTM 的 Caffe 模型转换为 CoreML。现在我正在尝试执行它。但是,我找不到处理整个序列的方法
np.ndarray( (7, #sequence
1, # batch
120, 1, 1)) #items dims
因为我找不到仅设置隐藏状态 (LSTM_1_c_in) 和初始历史记录 (LSTM_1_h_in) 的首字母的方法以及 自动在处理序列的下一项时使用先前的状态/结果。
它通过手动重启方法 'predict' 工作,并从以前的输出中手动设置 LSTM_1_c_in 和 LSTM_1_h_in(模型 reutrns LSTM_1_h_out 和 LSTM_1_c_out)。
是否可以通过 1 次运行处理 整个 序列?
附:使用 Swift 的方式也是可以接受的。
【问题讨论】:
标签: python swift neural-network coreml coremltools