github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-30 11:40:36 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

changed uhd arhi: you have to retrain uhd arhis from scratch.

Browse source
This commit is contained in:
Colombo 2020-05-06 19:21:23 +04:00
parent dcf380345b
commit 0c18b1f011
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
core/leras/archis/DeepFakeArchi.py
View file

@ -389,16 +389,14 @@ class DeepFakeArchi(nn.ArchiBase):
class ResidualBlock(nn.ModelBase):
def on_build(self, ch, kernel_size=3 ):
self.conv1 = nn.Conv2D( ch, ch*2, kernel_size=kernel_size, padding='SAME')
self.conv2 = nn.Conv2D( ch*2, ch, kernel_size=kernel_size, padding='SAME')
self.scale_add = nn.ScaleAdd(ch)
self.conv1 = nn.Conv2D( ch, ch, kernel_size=kernel_size, padding='SAME')
self.conv2 = nn.Conv2D( ch, ch, kernel_size=kernel_size, padding='SAME')
def forward(self, inp):
x = self.conv1(inp)
x = tf.nn.leaky_relu(x, 0.2)
x = self.conv2(x)
x = tf.nn.leaky_relu(x, 0.2)
x = self.scale_add([inp, x])
x = tf.nn.leaky_relu(inp + x, 0.2)
return x
class Encoder(nn.ModelBase):