4 ms·

I'm getting OOM errors on colab after the first 10-ish actions :( EDIT: Any chance you can support TPU on Colab?

• nickwalton00 7y ago
  Hmmm... I haven't seen that before there should be enough memory on the GPU to hold the model.
  • aharrison 7y ago
    Can corroborate, same problem of OOMs. Regardless, this is excellent work. Thank you for sharing.
    • Akababa 7y ago
      If it's any help, I restarted the notebook and haven't been getting crashes for a while now.