|
LBANN
0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
|
Phase specific "printf debugging". More...
#include <debug.hpp>
Public Member Functions | |
| debug (std::set< execution_mode > modes={}) | |
| Constructor. More... | |
| debug (const debug &)=default | |
| debug & | operator= (const debug &)=default |
| debug * | copy () const override |
| std::string | name () const override |
| Return this callback's name. More... | |
| void | on_batch_begin (model *m) override |
| Print that a batch is beginning. More... | |
| void | on_batch_end (model *m) override |
| Print that a batch is ending. More... | |
| void | on_batch_evaluate_begin (model *m) override |
| Print that a layer's forward prop is beginning. More... | |
| void | on_batch_evaluate_end (model *m) override |
| Print that a layer's forward prop is ending. More... | |
| void | on_forward_prop_begin (model *m, Layer *l) override |
| Print that a layer's forward prop is beginning. More... | |
| void | on_forward_prop_end (model *m, Layer *l) override |
| Print that a layer's forward prop is ending. More... | |
| void | on_backward_prop_begin (model *m, Layer *l) override |
| Print that a layer's backward prop is beginning. More... | |
| void | on_backward_prop_end (model *m, Layer *l) override |
| Print that a layer's backward prop is ending. More... | |
| void | on_evaluate_forward_prop_begin (model *m, Layer *l) override |
| Print that a layer's backward prop is beginning. More... | |
| void | on_evaluate_forward_prop_end (model *m, Layer *l) override |
| Print that a layer's backward prop is ending. More... | |
| void | on_optimize_begin (model *m, weights *w) override |
| Print that a weights' optimization step is beginning. More... | |
| void | on_optimize_end (model *m, weights *w) override |
| Print that a weights' optimization step is ending. More... | |
Serialization | |
| template<class Archive > | |
| void | serialize (Archive &ar) |
| Store state to archive for checkpoint and restart. More... | |
Public Member Functions inherited from lbann::callback_base | |
| callback_base (int batch_interval=1) | |
| Initialize a callback with an optional batch interval. More... | |
| callback_base (const callback_base &)=default | |
| virtual | ~callback_base ()=default |
| virtual void | setup (trainer *t) |
| Called once to set up the callback on the trainer. More... | |
| virtual void | setup (model *m) |
| Called once to set up the callback on the model (after all layers are set up). More... | |
| virtual void | on_setup_end (model *m) |
| Called at the end of setup. More... | |
| virtual void | on_train_begin (model *m) |
| Called at the beginning of training. More... | |
| virtual void | on_train_end (model *m) |
| Called at the end of training. More... | |
| virtual void | on_phase_end (model *m) |
| Called at the end of every phase (multiple epochs) in a layer-wise model training. More... | |
| virtual void | on_epoch_begin (model *m) |
| Called at the beginning of each epoch. More... | |
| virtual void | on_epoch_end (model *m) |
| Called immediate after the end of each epoch. More... | |
| virtual void | on_test_begin (model *m) |
| Called at the beginning of testing. More... | |
| virtual void | on_test_end (model *m) |
| Called immediately after the end of testing. More... | |
| virtual void | on_validation_begin (model *m) |
| Called at the beginning of validation. More... | |
| virtual void | on_validation_end (model *m) |
| Called immediately after the end of validation. More... | |
| virtual void | on_forward_prop_begin (model *m) |
| Called when a model begins forward propagation. More... | |
| virtual void | on_forward_prop_end (model *m) |
| Called when a model ends forward propagation. More... | |
| virtual void | on_backward_prop_begin (model *m) |
| Called when a model begins backward propagation. More... | |
| virtual void | on_backward_prop_end (model *m) |
| Called when a model ends backward propagation. More... | |
| virtual void | on_optimize_begin (model *m) |
| Called when a model begins optimization. More... | |
| virtual void | on_optimize_end (model *m) |
| Called when a model ends optimization. More... | |
| virtual void | on_evaluate_forward_prop_begin (model *m) |
| Called when a model begins forward propagation for evaluation (validation / testing). More... | |
| virtual void | on_evaluate_forward_prop_end (model *m) |
| Called when a model ends forward propagation for evaluation (validation / testing). More... | |
| int | get_batch_interval () const |
| Return the batch interval. More... | |
| virtual description | get_description () const |
| Human-readable description. More... | |
| template<class Archive > | |
| void | serialize (Archive &ar) |
| Store state to archive for checkpoint and restart. More... | |
| void | write_proto (lbann_data::Callback &proto) const |
| Write a protobuf description of the callback. More... | |
Private Member Functions | |
| void | write_specific_proto (lbann_data::Callback &proto) const final |
Private Attributes | |
| std::set< execution_mode > | m_modes |
| Execution modes for which status updates will be printed. More... | |
Additional Inherited Members | |
Protected Member Functions inherited from lbann::callback_base | |
| std::string | get_multi_trainer_path (const model &m, const std::string &root_dir) |
| Build a standard directory hierarchy including trainer ID. More... | |
| std::string | get_multi_trainer_ec_model_path (const model &m, const std::string &root_dir) |
| Build a standard directory hierachy including trainer, execution context, and model information (in that order). More... | |
| std::string | get_multi_trainer_model_path (const model &m, const std::string &root_dir) |
| Build a standard directory hierachy including trainer, model information in that order. More... | |
| callback_base & | operator= (const callback_base &)=default |
| Copy-assignment operator. More... | |
Protected Attributes inherited from lbann::callback_base | |
| int | m_batch_interval |
| Batch methods should once every this many steps. More... | |
Phase specific "printf debugging".
Print verbose status updates to standard error stream. This callback is useful for "printf debugging."
Takes a prototext parameter phase: train | validate | test | <empty> if <empty> will print messages for all phases
|
inline |
|
default |
|
inlineoverridevirtual |
Implements lbann::callback_base.
Definition at line 58 of file debug.hpp.
|
inlineoverridevirtual |
Return this callback's name.
Implements lbann::callback_base.
Definition at line 59 of file debug.hpp.
Print that a layer's backward prop is beginning.
Reimplemented from lbann::callback_base.
Print that a layer's backward prop is ending.
Reimplemented from lbann::callback_base.
|
overridevirtual |
Print that a batch is beginning.
Reimplemented from lbann::callback_base.
|
overridevirtual |
Print that a batch is ending.
Reimplemented from lbann::callback_base.
|
overridevirtual |
Print that a layer's forward prop is beginning.
Reimplemented from lbann::callback_base.
|
overridevirtual |
Print that a layer's forward prop is ending.
Reimplemented from lbann::callback_base.
|
overridevirtual |
Print that a layer's backward prop is beginning.
Reimplemented from lbann::callback_base.
Print that a layer's backward prop is ending.
Reimplemented from lbann::callback_base.
Print that a layer's forward prop is beginning.
Reimplemented from lbann::callback_base.
Print that a layer's forward prop is ending.
Reimplemented from lbann::callback_base.
Print that a weights' optimization step is beginning.
Reimplemented from lbann::callback_base.
Print that a weights' optimization step is ending.
Reimplemented from lbann::callback_base.
| void lbann::callback::debug::serialize | ( | Archive & | ar | ) |
Store state to archive for checkpoint and restart.
|
finalprivatevirtual |
Add callback specific data to prototext
Implements lbann::callback_base.
|
private |