Commit 0a2323f7 authored by Jan Dillenberger's avatar Jan Dillenberger
Browse files

increase randomness

parent 87dc53f6
......@@ -35,7 +35,7 @@ class EmbeddingCreator:
field_delim=' '
)
self.shuffled_batch = dataset.shuffle(128).padded_batch(batch_size, padded_shapes=((), ()), drop_remainder=True).prefetch(100)
self.shuffled_batch = dataset.shuffle(128000).padded_batch(batch_size, padded_shapes=((), ()), drop_remainder=True).prefetch(100)
self.dataset_iterator = self.shuffled_batch.make_initializable_iterator()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment