Thanks for bringing that up! I have been modifying a local copy of the training code and using Colab, and I just uploaded my changes to a fork on GitHub. Hopefully this will be more straightforward, but let me know if you have any issues using it:

https://github.com/GuitarML/Automated-GuitarAmpModelling