rztdl.dl.callbacks package

Submodules

rztdl.dl.callbacks.callback module

@created on: 1/3/20, @author: Himaprasoon, @version: v0.0.1

Description:

Sphinx Documentation Status:

class rztdl.dl.callbacks.callback.Callback(split: typing.Union[rztdl.dl.dataset.splits.DataSplit, str], frequency: int, interval_type: rztdl.dl.constants.string_constants.IntervalType, run_master_only=False)[source]

Bases: object

Parameters:
  • split (Union[DataSplit, str]) – Split on which callback has to be run
  • frequency (int) – No of Intervals to trigger callback
  • interval_type (IntervalType) – Interval Type => Batch /EPOCH etc
configure(*args, **kwargs)[source]
run(*args, **kwargs)[source]
set_split(split_handler)[source]

rztdl.dl.callbacks.callback_handler module

@created on: 12/20/19, @author: Himaprasoon, @version: v0.0.1

Description:

Sphinx Documentation Status:

class rztdl.dl.callbacks.callback_handler.CallbackHandler(run_callbacks=True)[source]

Bases: object

add(callback)[source]
Parameters:callback (Callback) – Callback Object
Returns:
initialize(flow)[source]
run(split_name, interval_type, metrics, counter)[source]

Runs a callback :type split_name: str :param split_name: name of split :type interval_type: IntervalType :param interval_type: IntervalType => Batch, epoch etc :type metrics: dict :param metrics: Dictionary with keys as metric names and metric values as values :type counter: int :param counter: Counter to check :return:

rztdl.dl.callbacks.checkpoint_callback module

@created on: 1/3/20, @author: Himaprasoon, @version: v0.0.1

Description:

Sphinx Documentation Status:

class rztdl.dl.callbacks.checkpoint_callback.CheckPointCallback(split: typing.Union[rztdl.dl.dataset.splits.DataSplit, str], frequency: int, interval_type: rztdl.dl.constants.string_constants.IntervalType, checkpoint_path: str, max_to_keep=3)[source]

Bases: rztdl.dl.callbacks.callback.Callback

Parameters:
  • split (Union[DataSplit, str]) – split
  • interval_type (IntervalType) – Interval Type : epoch/batch etc
  • frequency (int) – No of Intervals to trigger logging
  • max_to_keep – max checkpoints to keep
configure(flow, *args, **kwargs)[source]
run(counter, metrics, **kwargs)[source]

rztdl.dl.callbacks.log_callback module

@created on: 1/3/20, @author: Himaprasoon, @version: v0.0.1

Description:

Sphinx Documentation Status:

class rztdl.dl.callbacks.log_callback.LogCallback(split: typing.Union[rztdl.dl.dataset.splits.DataSplit, str], interval_type: rztdl.dl.constants.string_constants.IntervalType = None, frequency: int = None)[source]

Bases: rztdl.dl.callbacks.callback.Callback

Parameters:
  • split (Union[DataSplit, str]) – split
  • interval_type (Optional[IntervalType]) – Interval Type : epoch/batch etc
  • frequency (Optional[int]) – No of Intervals to trigger logging
configure(*args, **kwargs)[source]
run(counter, metrics, **kwargs)[source]

rztdl.dl.callbacks.tensorboard module

@created on: 1/3/20, @author: Himaprasoon, @version: v0.0.1

Description:

Sphinx Documentation Status:

class rztdl.dl.callbacks.tensorboard.TensorBoardMetricsCallback(split: typing.Union[rztdl.dl.dataset.splits.DataSplit, str], logdir: str, interval_type: rztdl.dl.constants.string_constants.IntervalType = <IntervalType.BATCH_END: 'BATCH_END'>, frequency: int = 1)[source]

Bases: rztdl.dl.callbacks.callback.Callback

Parameters:
  • split (Union[DataSplit, str]) – split
  • interval_type (IntervalType) – Interval Type : epoch/batch etc
  • frequency (int) – No of Intervals to trigger logging
configure(*args, **kwargs)[source]
run(counter, metrics, **kwargs)[source]

Module contents

@created on: 12/20/19, @author: Himaprasoon, @version: v0.0.1

Description:

Sphinx Documentation Status: