`main.py` and Google Colab: "Your session crashed after using all available RAM"

[sgbaird]

Using all 48322 compounds and the GPU hardware on Colab + the default batch size of 256, I get the following:

Your session crashed after using all available RAM. If you are interested in access to high-RAM runtimes, you may want to check out Colab Pro.

(switch dummy flag to False)

More of an FYI if someone else is trying to use Colab. Wouldn't be surprised if Colab Pro had enough RAM. How much RAM was available on the workstation that was used to run FTCP?

[sgbaird]

Maybe lowering the batch_size from 256 to e.g. 32 would help.

[sgbaird]

Update: reduced batch_size to 32 and same crash message.