Commit 0b76ec53 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): show plots

parent 82764754

zoo/detection/face.py

+2 −1
Original line number Diff line number Diff line
@@ -20,6 +20,7 @@ def train(train_cfg: str, session_name: str, level: str = 's', 
    model.train(

        data=train_cfg, epochs=max_epochs,

        name=session_name, project=_TRAIN_DIR,

        exist_ok=True, save=True, save_period=save_per_epoch, resume=resume,

        save=True, save_period=save_per_epoch, plots=True,

        exist_ok=True, resume=resume,

        **kwargs

    )

zoo/detection/person.py

+2 −1
Original line number Diff line number Diff line
@@ -20,6 +20,7 @@ def train(train_cfg: str, session_name: str, level: str = 's', 
    model.train(

        data=train_cfg, epochs=max_epochs,

        name=session_name, project=_TRAIN_DIR,

        exist_ok=True, save=True, save_period=save_per_epoch, resume=resume,

        save=True, save_period=save_per_epoch, plots=True,

        exist_ok=True, resume=resume,

        **kwargs

    )