GradientCallback

class radionets.core.GradientCallback(num_epochs, validation_data, arch_name, amp_phase)

Bases: Callback

Callback for gradient and prediction tracking.

Parameters:

num_epochsint

Number of training epochs.

validation_datastr or Path

Path to the validation dataset.

arch_namestr

Name of the architecture used for the model.

amp_phasebool

Whether to use amplitude-phase representation.

Methods Summary

after_cancel_backward()
after_epoch()
before_backward()

Methods Documentation

after_cancel_backward()

after_epoch()

before_backward()