Skip to content

Commit

Permalink
Update detection configuration: increase warmup steps and patience, a…
Browse files Browse the repository at this point in the history
…dd min_lr, and remove unused callbacks
  • Loading branch information
eugene123tw committed Jan 15, 2025
1 parent aa148ac commit 6039c95
Showing 1 changed file with 3 additions and 18 deletions.
21 changes: 3 additions & 18 deletions src/otx/recipe/detection/dfine_x_tile.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -15,14 +15,15 @@ model:
scheduler:
class_path: otx.core.schedulers.LinearWarmupSchedulerCallable
init_args:
num_warmup_steps: 5
num_warmup_steps: 20
main_scheduler_callable:
class_path: lightning.pytorch.cli.ReduceLROnPlateau
init_args:
mode: max
factor: 0.1
patience: 6
patience: 9
monitor: val/map_50
min_lr: 2e-06
engine:
task: DETECTION
device: auto
Expand All @@ -31,22 +32,6 @@ callback_monitor: val/map_50

data: ../_base_/data/detection_tile.yaml
overrides:
callbacks:
- class_path: otx.algo.callbacks.adaptive_train_scheduling.AdaptiveTrainScheduling
init_args:
max_interval: 1
decay: -0.025
min_lrschedule_patience: 3
- class_path: otx.algo.callbacks.adaptive_early_stopping.EarlyStoppingWithWarmup
init_args:
monitor: null
mode: max
patience: 10
check_on_train_epoch_end: false
min_delta: 0.001
warmup_iters: 100
warmup_epochs: 7

reset:
- data.train_subset.transforms
- data.val_subset.transforms
Expand Down

0 comments on commit 6039c95

Please sign in to comment.