Skip to content

Commit 069b9ef

Browse files
authored
Fixed focal loss not beeing optimized by autotune (#759)
1 parent a78c260 commit 069b9ef

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

birdnet_analyzer/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -785,7 +785,7 @@ def lr_schedule(epoch, lr):
785785
optimizer_cls = keras.optimizers.legacy.Adam if sys.platform == "darwin" else keras.optimizers.Adam
786786

787787
def _focal_loss(y_true, y_pred):
788-
return focal_loss(y_true, y_pred, gamma=cfg.FOCAL_LOSS_GAMMA, alpha=cfg.FOCAL_LOSS_ALPHA)
788+
return focal_loss(y_true, y_pred, gamma=focal_loss_gamma, alpha=focal_loss_alpha)
789789

790790
# Choose the loss function based on config
791791
loss_function = _focal_loss if train_with_focal_loss else custom_loss

0 commit comments

Comments
 (0)