[Task]: The number of epoch is 999 when using EfficentAD I want to change this.

[tamamoto0821]

What is the motivation for this task?

The number of epoch is 999 when using EfficentAD
I want to change this.

Describe the solution you'd like

The number of epoch is 999 when using EfficentAD
I want to change this.

Additional context

The number of epoch is 999 when using EfficentAD
I want to change this.

[samet-akcay]

Are you using CLI or API?

you could set max_epochs argument from the Engine.

CLI

anomalib train --model anomalib.models.EfficientAd --data MVTec --trainer.max_epochs 3

API

from anomalib.data import MVTec
from anomalib.models import EfficientAd
from anomalib.engine import Engine

datamodule = MVTec()
model = EfficientAd()
engine = Engine(max_epochs=10)

engine.train(datamodule=datamodule, model=model)