Skip to content

Commit 3cf1821

Browse files
committed
Fix incorrect docstrings
Signed-off-by: Samet Akcay <[email protected]>
1 parent ecba6f8 commit 3cf1821

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

src/anomalib/models/image/padim/lightning_model.py

+10-3
Original file line numberDiff line numberDiff line change
@@ -84,14 +84,21 @@ class Padim(MemoryBankMixin, AnomalibModule):
8484
result images. Defaults to ``True``.
8585
8686
Example:
87-
>>> from anomalib.models.image.padim import Padim
87+
>>> from anomalib.models import Padim
88+
>>> from anomalib.data import MVTec
89+
>>> from anomalib.engine import Engine
90+
91+
>>> # Initialize model and data
92+
>>> datamodule = MVTec()
8893
>>> model = Padim(
8994
... backbone="resnet18",
9095
... layers=["layer1", "layer2", "layer3"],
9196
... pre_trained=True
9297
... )
93-
>>> model.fit()
94-
>>> prediction = model(image)
98+
99+
>>> engine = Engine()
100+
>>> engine.train(model=model, datamodule=datamodule)
101+
>>> predictions = engine.predict(model=model, datamodule=datamodule)
95102
96103
Note:
97104
The model does not require training in the traditional sense. It fits

0 commit comments

Comments
 (0)