GradientCallback#

class radionets.core.callbacks.GradientCallback(num_epochs, test_data, arch_name, amp_phase)[source]#

Bases: Callback

Attributes Summary

learn

name

Name of the Callback, camel-cased and with 'Callback' removed

order

run

run_train

run_valid

Methods Summary

__call__(event_name)

Call self.{event_name} if it's defined

after_cancel_backward()

after_epoch()

before_backward()

Attributes Documentation

learn = None#
name#

Name of the Callback, camel-cased and with ‘Callback’ removed

order = 0#
run = True#
run_train = True#
run_valid = True#

Methods Documentation

__call__(event_name)#

Call self.{event_name} if it’s defined

after_cancel_backward()[source]#
after_epoch()[source]#
before_backward()[source]#