Example script failure in Colab

Just wishing to verify my issue:

When I copy the example script straight from github, and run it in Colab Pro (25 GB of RAM), the example_model.py fails as the instance crashes due to RAM. See here: https://prnt.sc/1uddvgy

Is it that I need even more RAM? or do I have another issue?

Thanks in advance.

First, you can try using the int8 data files which use less RAM:


You might still need more RAM though. I think my model uses somewhere between 28 to 38 GB of memory.