Unverified 提交 b52fd48b authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

TFDWConv() `depthwise_initializer` fix (#7845)

上级 05d4fc9d
...@@ -97,7 +97,7 @@ class TFDWConv(keras.layers.Layer): ...@@ -97,7 +97,7 @@ class TFDWConv(keras.layers.Layer):
strides=s, strides=s,
padding='SAME' if s == 1 else 'VALID', padding='SAME' if s == 1 else 'VALID',
use_bias=not hasattr(w, 'bn'), use_bias=not hasattr(w, 'bn'),
kernel_initializer=keras.initializers.Constant(w.conv.weight.permute(2, 3, 1, 0).numpy()), depthwise_initializer=keras.initializers.Constant(w.conv.weight.permute(2, 3, 1, 0).numpy()),
bias_initializer='zeros' if hasattr(w, 'bn') else keras.initializers.Constant(w.conv.bias.numpy())) bias_initializer='zeros' if hasattr(w, 'bn') else keras.initializers.Constant(w.conv.bias.numpy()))
self.conv = conv if s == 1 else keras.Sequential([TFPad(autopad(k, p)), conv]) self.conv = conv if s == 1 else keras.Sequential([TFPad(autopad(k, p)), conv])
self.bn = TFBN(w.bn) if hasattr(w, 'bn') else tf.identity self.bn = TFBN(w.bn) if hasattr(w, 'bn') else tf.identity
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论