通过将权重(PWC-Net)从pytorch模型转移到tf.keras模型来实现输出差异

时间:2019-09-04 15:25:09

标签: pytorch tf.keras opticalflow transfer-learning

我参考此回购协议

@bazza
在tf keras中实现了pwc网络。我用来初始化pwc net的conv层的权重的权重与回购中pytorch模型的权重相同。我多次检查了我的keras模型,确保我的模型与pytorchmodel相同。所以我想在pytorch和tf keras 2.0之间有权重转移的问题吗?

the result is

0 个答案:

没有答案