Yeah, I'm currently playing with that entire repository of models. Unfortunately the ones that exist in this package don't seem to fully implement their features as far as I can tell? Pulling the individual models apart with their own data loading has been... a chore.
07/07/2023, 5:35 PM
they do work but you have either convert them to pytorch lightning + make your own data loaders, or try to make your dataset conform to theirs
found it a lot easier to just re-build the networks in PL + generate our own dataloaders