Skip to content
Snippets Groups Projects
Commit c7901791 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by tensorflow-copybara
Browse files

Fix a bug when loading Shakespeare dataset.

The dataset returns only a single element and cannot be unpacked.

PiperOrigin-RevId: 322454423
parent 1c790ebc
No related branches found
No related tags found
No related merge requests found
......@@ -74,9 +74,10 @@ def main(argv):
if len(argv) > 1:
raise app.UsageError('Too many command-line arguments.')
train_clientdata, test_dataset = shakespeare_dataset.construct_character_level_datasets(
train_clientdata = shakespeare_dataset.construct_character_level_datasets(
FLAGS.client_batch_size, FLAGS.client_epochs_per_round,
FLAGS.sequence_length)
_, test_dataset = shakespeare_dataset.construct_centralized_datasets()
test_dataset = test_dataset.cache()
loss_fn_builder = functools.partial(
......
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