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 449
449
" # on each TPU worker.\n " ,
450
450
" per_replica_batch_size = batch_size // strategy.num_replicas_in_sync\n " ,
451
451
" \n " ,
452
- " train_dataset = strategy.experimental_distribute_datasets_from_function (\n " ,
452
+ " train_dataset = strategy.distribute_datasets_from_function (\n " ,
453
453
" lambda _: get_dataset(per_replica_batch_size, is_training=True))\n " ,
454
454
" \n " ,
455
455
" @tf.function\n " ,
588
588
"metadata" : {
589
589
"accelerator" : " TPU" ,
590
590
"colab" : {
591
- "collapsed_sections" : [],
592
591
"name" : " tpu.ipynb" ,
593
- "toc_visible" : true
592
+ "toc_visible" : true ,
593
+ "provenance" : []
594
594
},
595
595
"kernelspec" : {
596
596
"display_name" : " Python 3" ,
599
599
},
600
600
"nbformat" : 4 ,
601
601
"nbformat_minor" : 0
602
- }
602
+ }
You can’t perform that action at this time.
0 commit comments