Kaynağa Gözat

TFDWConv() `depthwise_initializer` fix (#7845)

modifyDataloader
Glenn Jocher GitHub 2 yıl önce
ebeveyn
işleme
b52fd48b4a
Veri tabanında bu imza için bilinen anahtar bulunamadı GPC Anahtar Kimliği: 4AEE18F83AFDEB23
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  1. +1
    -1
      models/tf.py

+ 1
- 1
models/tf.py Dosyayı Görüntüle

@@ -97,7 +97,7 @@ class TFDWConv(keras.layers.Layer):
strides=s,
padding='SAME' if s == 1 else 'VALID',
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()))
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

Yükleniyor…
İptal
Kaydet