❓ What should I do if I want to save weights every 5 cycles? Thanks

[shz1314]

What is the motivation for this task?

What should I do if I want to save weights every 5 cycles? Thanks

Describe the solution you'd like

What should I do if I want to save weights every 5 cycles? Thanks

Additional context

What should I do if I want to save weights every 5 cycles? Thanks

[blaz-r]

You will need to modify the following code:

anomalib/src/anomalib/utils/callbacks/__init__.py

Lines 63 to 69 in 1f50c95

	checkpoint = ModelCheckpoint(
	dirpath=os.path.join(config.project.path, "weights", "lightning"),
	filename="model",
	monitor=monitor_metric,
	mode=monitor_mode,
	auto_insert_metric_name=False,
	)

and add every_n_epochs or every_n_train_steps or whichever suits you. You might need to remove monitor, but I am not sure. Check ModelCheckpoint page for more details on checkpointing.