We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ec76585 commit 6879b09Copy full SHA for 6879b09
site/en/guide/tpu.ipynb
@@ -449,7 +449,7 @@
449
"# on each TPU worker.\n",
450
"per_replica_batch_size = batch_size // strategy.num_replicas_in_sync\n",
451
"\n",
452
- "train_dataset = strategy.experimental_distribute_datasets_from_function(\n",
+ "train_dataset = strategy.distribute_datasets_from_function(\n",
453
" lambda _: get_dataset(per_replica_batch_size, is_training=True))\n",
454
455
"@tf.function\n",
@@ -588,9 +588,9 @@
588
"metadata": {
589
"accelerator": "TPU",
590
"colab": {
591
- "collapsed_sections": [],
592
"name": "tpu.ipynb",
593
- "toc_visible": true
+ "toc_visible": true,
+ "provenance": []
594
},
595
"kernelspec": {
596
"display_name": "Python 3",
@@ -599,4 +599,4 @@
599
600
"nbformat": 4,
601
"nbformat_minor": 0
602
-}
+}
0 commit comments