【发布时间】:2018-03-26 12:57:32
【问题描述】:
我正在尝试使用 DL4J 将预训练的 Keras 模型加载到 Java 应用程序中。 在 Keras 中训练时,我使用了“channels_first”image_data_format,现在在 DL4J 中我无法正确加载我的模型,因为它尝试使用“channels_last”配置。
DL4J 中是否有与 Keras 中的 set_image_data_format("channels_first) 类似的功能?
【问题讨论】: