Comment by Akababa

7 years ago

I'm getting OOM errors on colab after the first 10-ish actions :(

EDIT: Any chance you can support TPU on Colab?

3 comments

Akababa

Reply
nickwalton00  7 years ago

Hmmm... I haven't seen that before there should be enough memory on the GPU to hold the model.

  • aharrison  7 years ago

    Can corroborate, same problem of OOMs. Regardless, this is excellent work. Thank you for sharing.

    • Akababa  7 years ago

      If it's any help, I restarted the notebook and haven't been getting crashes for a while now.