scripts/evaluate.py

import argparse

import pytorch_lightning as pl

from core.memfof_lit import MEMFOFLit, DataModule
from config.parser import parse_args


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--cfg", help="experiment configure file name", required=True, type=str
    )
    args = parse_args(parser)

    trainer = pl.Trainer(
        accelerator="auto",
    )

    datamodule = DataModule(args)
    model = MEMFOFLit(args)
    trainer.validate(model, datamodule)