SwitchLoss

class radionets.core.SwitchLoss(second_loss, when_switch)

Bases: Callback

Callback for switching loss functions during training.

Changes the loss function to a different one after a specified number of epochs.

Parameters:

second_losscallable

The loss function to switch to.

when_switchint

Epoch number after which to switch loss functions.

Methods Summary

before_epoch()

Methods Documentation

before_epoch()