CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
iperov
2021-06-10 14:42:58 +04:00
parent 9f0b4bf3cf
commit 62c6fffdde
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/leras/layers/Conv2D.py
+1 -1
View File
@@ -98,7 +98,7 @@ class Conv2D(nn.LayerBase):
else: else:
dilations = [1,1,dilations,dilations] dilations = [1,1,dilations,dilations]
x = tf.nn.conv2d(x, weight, self.strides, 'VALID', dilations=self.dilations, data_format=nn.data_format) x = tf.nn.conv2d(x, weight, strides, 'VALID', dilations=dilations, data_format=nn.data_format)
if self.use_bias: if self.use_bias:
if nn.data_format == "NHWC": if nn.data_format == "NHWC":
bias = tf.reshape (self.bias, (1,1,1,self.out_ch) ) bias = tf.reshape (self.bias, (1,1,1,self.out_ch) )