niftynet.contrib.learning_rate_schedule.decay_lr_application module

class DecayLearningRateApplication(net_param, action_param, is_training)[source]

Bases: niftynet.application.segmentation_application.SegmentationApplication

REQUIRED_CONFIG_SECTION = 'SEGMENTATION'
connect_data_and_network(outputs_collector=None, gradients_collector=None)[source]

Adding sampler output tensor and network tensors to the graph.

Parameters:
  • outputs_collector
  • gradients_collector
Returns:

set_iteration_update(iteration_message)[source]

This function will be called by the application engine at each iteration.