Hi @acl21,
Thanks for sharing this amazing repository! I have tried to train your default code (on both main and dev branches), but was unable to obtain good results. Here is a screenshot of some of the logged training curves.

You may also find other training data and videos here:
https://drive.google.com/drive/folders/1xP1IpPDGsw4CwjiPVpauKRN_CL0smQHQ?usp=sharing
Warm Regards,
Derek