LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
lbann::KFAC Member List

This is the complete list of members for lbann::KFAC, including all inherited members.

allgather_precondition_gradient(lbann_comm &comm, ExeContextType &context)lbann::KFACprivate
apply(ExecutionContext &context, model &m, data_coordinator &dc, execution_mode mode) finallbann::KFACvirtual
lbann::TrainingAlgorithm::apply(model &model, data_coordinator &dc)lbann::TrainingAlgorithminline
damping_0_defaultlbann::KFACstatic
damping_warmup_steps_defaultlbann::KFACstatic
Devicelbann::KFACstatic
do_batch_begin_cbs(model &model)lbann::KFACprotected
do_batch_end_cbs(model &model)lbann::KFACprotected
do_epoch_begin_cbs(model &model)lbann::KFACprotected
do_epoch_end_cbs(model &model)lbann::KFACprotected
do_get_new_execution_context() const finallbann::KFACprotectedvirtual
do_train_begin_cbs(model &model)lbann::KFACprotected
do_train_end_cbs(model &model)lbann::KFACprotected
end_old_async_weights_model(model &model, lbann_comm *comm, ExeContextType &context)lbann::KFACprivate
end_send_recv_inverse_matrices(ExeContextType &context, lbann_comm *comm)lbann::KFACprotected
end_sync_weights_async(model &model, lbann_comm *comm)lbann::KFACprivate
ExeContextType typedeflbann::KFAC
get_name() const noexceptlbann::TrainingAlgorithm
get_new_execution_context() constlbann::TrainingAlgorithminline
get_type() const finallbann::KFACvirtual
KFAC(std::string name, std::unique_ptr< TermCriteriaType > stop, std::vector< double > damping_act_params, std::vector< double > damping_err_params, std::vector< double > damping_bn_act_params, std::vector< double > damping_bn_err_params, std::vector< bool > kfac_use_interval, size_t damping_warmup_steps, double kronecker_decay, bool print_time, bool print_matrix, bool print_matrix_summary, bool use_pi, std::vector< size_t > update_intervals, size_t update_interval_steps, kfac::kfac_inverse_strategy inverse_strategy, std::vector< std::string > disable_layers, double learning_rate_factor, double learning_rate_factor_gru, size_t compute_interval, bool distribute_precondition_compute, bool use_eigen_decomposition, bool enable_copy_errors, bool enable_copy_activations)lbann::KFAC
KFAC(KFAC const &other)=deletelbann::KFAC
KFAC(KFAC &&other)=defaultlbann::KFAC
kronecker_decay_defaultlbann::KFACstatic
m_compute_intervallbann::KFACprivate
m_damping_act_paramslbann::KFACprivate
m_damping_bn_act_paramslbann::KFACprivate
m_damping_bn_err_paramslbann::KFACprivate
m_damping_err_paramslbann::KFACprivate
m_damping_warmup_stepslbann::KFACprivate
m_disable_layerslbann::KFACprivate
m_distribute_precondition_computelbann::KFACprivate
m_enable_copy_activationslbann::KFACprivate
m_enable_copy_errorslbann::KFACprivate
m_global_inverse_buffer_sizelbann::KFACprivate
m_has_kronecker_inverselbann::KFACprivate
m_inverse_matrices_sizelbann::KFACprivate
m_inverse_matrix_communication_reqslbann::KFACprivate
m_inverse_strategylbann::KFACprivate
m_kronecker_decaylbann::KFACprivate
m_learning_rate_factorlbann::KFACprivate
m_learning_rate_factor_grulbann::KFACprivate
m_print_matrixlbann::KFACprivate
m_print_matrix_summarylbann::KFACprivate
m_print_timelbann::KFACprivate
m_stopping_criterialbann::KFACprivate
m_time_backward_passlbann::KFACprivate
m_time_forward_passlbann::KFACprivate
m_time_kfaclbann::KFACprivate
m_time_span_backward_commlbann::KFACprivate
m_time_span_backward_comm_endlbann::KFACprivate
m_time_span_forward_commlbann::KFACprivate
m_time_span_forward_comm_endlbann::KFACprivate
m_time_span_inverse_commlbann::KFACprivate
m_time_span_inverse_send_recvlbann::KFACprivate
m_time_span_precond_commlbann::KFACprivate
m_update_interval_stepslbann::KFACprivate
m_update_intervalslbann::KFACprivate
m_use_eigen_decompositionlbann::KFACprivate
m_use_KFAC_epochlbann::KFACprivate
m_use_pilbann::KFACprivate
m_weight_matrices_buffer_sizelbann::KFACprivate
m_weights_communication_reqslbann::KFACprivate
on_backward_prop_end(ExeContextType &context, model &model)lbann::KFACprivate
on_forward_prop_end(ExeContextType &context, model &model)lbann::KFACprivate
operator=(const KFAC &other)=deletelbann::KFAC
operator=(KFAC &&other)=defaultlbann::KFAC
lbann::TrainingAlgorithm::operator=(const TrainingAlgorithm &other)=deletelbann::TrainingAlgorithmprotected
lbann::TrainingAlgorithm::operator=(TrainingAlgorithm &&other)=defaultlbann::TrainingAlgorithmprotected
prof_colorlbann::KFACstatic
prof_synclbann::KFACstatic
setup_models(std::vector< observer_ptr< model >> const &models, size_t max_mini_batch_size, const std::vector< El::Grid *> &grids)lbann::TrainingAlgorithm
start_old_async_weights_model(model &model, lbann_comm *comm, ExeContextType &context)lbann::KFACprivate
start_send_recv_inverse_matrices(ExeContextType &context, lbann_comm *comm)lbann::KFACprotected
start_sync_weights_async(model &model, lbann_comm *comm)lbann::KFACprivate
sync_weights_model(model &model, lbann_comm *comm)lbann::KFACprivate
TermCriteriaType typedeflbann::KFAC
train(ExeContextType &c, model &model, data_coordinator &dc, TermCriteriaType const &term)lbann::KFAC
train_mini_batch(ExeContextType &c, model &model, data_coordinator &dc)lbann::KFACprotected
TrainingAlgorithm(std::string name)lbann::TrainingAlgorithm
TrainingAlgorithm(const TrainingAlgorithm &other)=deletelbann::TrainingAlgorithmprotected
TrainingAlgorithm(TrainingAlgorithm &&other)=defaultlbann::TrainingAlgorithmprotected
~KFAC() noexcept=defaultlbann::KFAC
~TrainingAlgorithm()=defaultlbann::TrainingAlgorithmvirtual