github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 13:32:09 -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
parent c90d67bec1
commit 57fd6735a7
3 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
facelib/LandmarksExtractor.py
View file

@ -57,7 +57,7 @@ class LandmarksExtractor(object):
def __init__ (self, keras):
self.keras = keras
K = self.keras.backend
class TorchBatchNorm2D(self.keras.engine.topology.Layer):
class TorchBatchNorm2D(self.keras.layers.Layer):
def __init__(self, axis=-1, momentum=0.99, epsilon=1e-3, **kwargs):
super(TorchBatchNorm2D, self).__init__(**kwargs)
self.supports_masking = True