Skip to content

Commit

Permalink
SSIM eval callback for continuous eval
Browse files Browse the repository at this point in the history
Summary:
Eval Callback design #3 from https://docs.google.com/document/d/1-6HUrsO6pcXW1HDhqowdz_DWziXeGacDekg1BublY_c/edit#heading=h.wrjehs3xh1h4

This diff implements utils and SSIM eval.

Supersedes D50397800

Differential Revision: D50823825

fbshipit-source-id: d9401b24cb3013e4de5edfa70fe78309ef261596
  • Loading branch information
dongyuzheng authored and facebook-github-bot committed Nov 2, 2023
1 parent f0823a9 commit 25fa5ac
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion torchtnt/framework/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
from .evaluate import evaluate
from .fit import fit
from .predict import predict
from .state import PhaseState, State
from .state import ActivePhase, EntryPoint, PhaseState, State
from .train import train
from .unit import EvalUnit, PredictUnit, TEvalUnit, TPredictUnit, TrainUnit, TTrainUnit

Expand All @@ -20,6 +20,8 @@
"evaluate",
"fit",
"predict",
"ActivePhase",
"EntryPoint",
"PhaseState",
"State",
"train",
Expand Down

0 comments on commit 25fa5ac

Please sign in to comment.