Skip to main content
https://github.com/MEPP-team/SpecTrHuMS
28 July 2023, 12:42:45 UTC
Raw File Download Save again
Tip revision: 683831be6abc326522c36c34160f76a4bc36a9dd authored by Clément Lemeunier on 26 July 2023, 06:49:28 UTC
README.md
Tip revision: 683831b
train.py 
import learning.trainer as lt

if __name__ == "__main__":
    trainer, opt = lt.load_trainer(load=False)

    trainer.fit(opt["model"], opt["dataloader_train"], opt["dataloader_test"])