neuroailab / tfutils

Utilities for working with tensorflow
MIT License
25 stars 8 forks source link

Adds functionality to train on TPU pods or slices #151

Closed jvrsgsty closed 4 years ago

jvrsgsty commented 4 years ago

Now you can specify a tpu pod or slice in tpu_name, but will require an additional single device eval_tpu_name to be specified to perform evaluation on, as eval is currently unsupported on pods.

Also accounts for different locations of the objects required for TPUTraining: