dmlcloud.Stage

class dmlcloud.Stage(name=None, epochs=1)

Hook Points:

• pre_stage()

• post_stage()

• pre_epoch()

• post_epoch()

__init__(name=None, epochs=1)

Methods

__init__([name, epochs])
add_callback(callback[, priority])	Adds a callback to this stage.
add_column(name[, metric, formatter, width, ...])	Adds a column to the table.
add_metric(name, metric)
enable_profiler([epochs, schedule])	Enables the profiler for this stage.
finish_step()
log(name, value[, reduction, prefixed, log_step])
next_epoch()	Advances the stage to the next epoch.
post_epoch()	Executed after each epoch.
post_stage()	Executed after the stage finishes.
post_step()	Executed after each step.
pre_epoch()	Executed before each epoch.
pre_stage()	Executed before the stage starts.
run()	Override this method to implement the main logic of the stage and do manual epoch management.
run_epoch()	Override this method to implement the main logic of the stage for a single epoch.

Attributes

config	Same as Pipeline.config.
current_epoch
device	Same as Pipeline.device.
end_time
epoch_end_time
epoch_start_time
has_profiler	Returns True if the profiler is enabled for this stage, otherwise False.
profiler	If enabled, returns the profiler object associated with this stage, otherwise None.
run_dir	Same as Pipeline.run_dir.
start_time
table