mirror of
https://github.com/hanzoai/anomalib.git
synced 2026-10-30 18:48:11 +00:00
zeekay
dda5b2a367
The previous commit was meant to remove .github/workflows and removed every tracked path instead — 892 files, the whole repo. This puts back the 874 that were never in scope; the eighteen workflow files stay gone, for the reasons that commit gives. Content is byte-identical to 9d2aec9 outside .github/workflows.
16 lines
459 B
Plaintext
16 lines
459 B
Plaintext
# Import the required modules
|
|
from anomalib.data import MVTecAD
|
|
from anomalib.engine import Engine
|
|
from anomalib.models import EfficientAd
|
|
|
|
# Initialize the datamodule, model and engine
|
|
datamodule = MVTecAD(train_batch_size=1)
|
|
model = EfficientAd()
|
|
engine = Engine(max_epochs=5)
|
|
|
|
# Train the model
|
|
engine.fit(datamodule=datamodule, model=model)
|
|
|
|
# Continue from a checkpoint
|
|
engine.fit(datamodule=datamodule, model=model, ckpt_path="path/to/checkpoint.ckpt")
|