Skip to content

Commit 669b7f1

Browse files
Return metrics from train (#2702)
Signed-off-by: Ashwin Vaidya <ashwinnitinvaidya@gmail.com>
1 parent e6da810 commit 669b7f1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/anomalib/engine/engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -721,7 +721,7 @@ def train(
721721
self.trainer.validate(model, val_dataloaders, None, verbose=False, datamodule=datamodule)
722722
else:
723723
self.trainer.fit(model, train_dataloaders, val_dataloaders, datamodule, ckpt_path)
724-
self.trainer.test(model, test_dataloaders, ckpt_path=ckpt_path, datamodule=datamodule)
724+
return self.trainer.test(model, test_dataloaders, ckpt_path=ckpt_path, datamodule=datamodule)
725725

726726
def export(
727727
self,

0 commit comments

Comments
 (0)