loss_on_threshold_with_trainer_state.yaml

controller_metrics:
  - name: trainer_state
    class: TrainingState
  - name: training_loss
    class: Loss
controllers:
  - name: loss_controller
    triggers:
      - on_log
    rule: training_loss['loss'] < 2 and trainer_state["epoch"] >= 0.5
    operations:
      - hfcontrols.should_training_stop