fix a bug, load_weights doesn't return anything
@@ -192,7 +192,7 @@ if __name__ == '__main__':
show_shapes=True)
if args.weights:
- vae = vae.load_weights(args.weights)
+ vae.load_weights(args.weights)
else:
# train the autoencoder
vae.fit(x_train,
