text
text copied to clipboard
Bug fix - Non TPU env
If TPU acceleration is not available, os.environ['COLAB_TPU_ADDR'] fails (keyError). Fixed it to os.environ.get('COLAB_TPU_ADDR'), which returns NoneType and then continue to look for GPU or raise ValueError
Check out this pull request on
See visual diffs & provide feedback on Jupyter Notebooks.
Powered by ReviewNB