This model is just a prototype used during the development process of GelGenie. It consists of a SMP UNet model trained on the LSDB component of the GelGenie dataset. More details in the config file.

Our codebase is fully open-sourced and is available here: https://github.com/mattaq31/GelGenie.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Examples
Inference API (serverless) does not yet support pytorch models for this pipeline type.