Skip to content

Commit 8eb3ae9

Browse files
authored
Fix UFLow configuration (#4504)
add callbacks for uflow
1 parent f225bb3 commit 8eb3ae9

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/otx/recipe/anomaly/uflow.yaml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,13 @@ callback_monitor: image_F1Score
1717
data: ../_base_/data/anomaly.yaml
1818

1919
callbacks:
20+
- class_path: otx.backend.native.callbacks.adaptive_early_stopping.EarlyStoppingWithWarmup
21+
init_args:
22+
patience: 5
23+
mode: max
24+
- class_path: otx.backend.native.callbacks.adaptive_train_scheduling.AdaptiveTrainScheduling
25+
init_args:
26+
max_interval: 1
2027
- class_path: lightning.pytorch.callbacks.ModelCheckpoint
2128
init_args:
2229
dirpath: "" # use engine.work_dir

0 commit comments

Comments
 (0)