Glenn Jocher
GitHub
2 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
models/tf.py
View File
| 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 |
Loading…