Skip to content
Snippets Groups Projects

VAE : I'll get him anyway!

Former-commit-id: 028b4797
parent d1366b13
No related branches found
No related tags found
No related merge requests found
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
...@@ -106,7 +106,7 @@ class DataGenerator(Sequence): ...@@ -106,7 +106,7 @@ class DataGenerator(Sequence):
def on_epoch_end(self): def on_epoch_end(self):
self.cluster_i = clusters_size self.cluster_i = self.clusters_size
self.read_next_cluster() self.read_next_cluster()
......
...@@ -234,8 +234,8 @@ class VariationalAutoencoder(): ...@@ -234,8 +234,8 @@ class VariationalAutoencoder():
shuffle = True, shuffle = True,
epochs = epochs, epochs = epochs,
initial_epoch = initial_epoch, initial_epoch = initial_epoch,
callbacks = callbacks_list callbacks = [], #callbacks_list,
# validation_data = (x_test, x_test) validation_data = (x_test, x_test)
) )
end_time = time.time() end_time = time.time()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment