https://github.com/MEPP-team/SpecTrHuMS
Tip revision: 683831be6abc326522c36c34160f76a4bc36a9dd authored by Clément Lemeunier on 26 July 2023, 06:49:28 UTC
README.md
README.md
Tip revision: 683831b
test.py
import learning.trainer as lt
import pytorch_lightning as pl
import os
from torch.multiprocessing import set_start_method
if __name__ == "__main__":
trainer, opt = lt.load_trainer(load=True)
trainer.test(
opt["model"],
opt["dataloader_test"],
ckpt_path=opt["best_checkpoint_filename"],
)
