Are there pretrained weights available?
Closed this issue · 1 comments
jamesbraza commented
GEARS.load_pretrained loads in a model.pt to GEARS.model.
Where can one find that model.pt file? Is there a pre-trained variant available somewhere (e.g. Hugging Face)?
yhr91 commented
Yes, there is a pretrained model available through this Colab notebook