github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-20 21:43:21 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

update to Keras 2.2.2

Browse source
This commit is contained in:
TooMuchFun 2018-11-14 07:22:27 -08:00
commit 57fd6735a7
3 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
nnlib/__init__.py
View file

@ -84,10 +84,10 @@ def MSEMaskLossClass(keras):
return MSEMaskLoss
def PixelShufflerClass(keras):
class PixelShuffler(keras.engine.topology.Layer):
class PixelShuffler(keras.layers.Layer):
def __init__(self, size=(2, 2), data_format=None, **kwargs):
super(PixelShuffler, self).__init__(**kwargs)
self.data_format = keras.utils.conv_utils.normalize_data_format(data_format)
self.data_format = keras.backend.common.normalize_data_format(data_format)
self.size = keras.utils.conv_utils.normalize_tuple(size, 2, 'size')
def call(self, inputs):