File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 449449 " # on each TPU worker.\n " ,
450450 " per_replica_batch_size = batch_size // strategy.num_replicas_in_sync\n " ,
451451 " \n " ,
452- " train_dataset = strategy.experimental_distribute_datasets_from_function (\n " ,
452+ " train_dataset = strategy.distribute_datasets_from_function (\n " ,
453453 " lambda _: get_dataset(per_replica_batch_size, is_training=True))\n " ,
454454 " \n " ,
455455 " @tf.function\n " ,
588588 "metadata" : {
589589 "accelerator" : " TPU" ,
590590 "colab" : {
591- "collapsed_sections" : [],
592591 "name" : " tpu.ipynb" ,
593- "toc_visible" : true
592+ "toc_visible" : true ,
593+ "provenance" : []
594594 },
595595 "kernelspec" : {
596596 "display_name" : " Python 3" ,
599599 },
600600 "nbformat" : 4 ,
601601 "nbformat_minor" : 0
602- }
602+ }
You can’t perform that action at this time.
0 commit comments