raduciou

that's my first comment on github so: you need to put name of weights in order to save model (keras bug) `self.add_weight(shape=(1, 1, 1, 512), dtype=tf.float32, initializer=tf.keras.initializers.Ones(), trainable=True, name='fckr')` after...

got the same problem. training on 1000 pics and for 30 epochs.