|
|
|
@ -29,6 +29,8 @@ class CallbackParam {
|
|
|
|
|
CallbackParam(int64_t epoch_num, int64_t cur_epoch_step, int64_t total_step_num)
|
|
|
|
|
: cur_epoch_num_(epoch_num), cur_epoch_step_num_(cur_epoch_step), cur_step_num_(total_step_num) {}
|
|
|
|
|
|
|
|
|
|
~CallbackParam() = default;
|
|
|
|
|
|
|
|
|
|
// these are constant public fields for easy access and consistency with python cb_param
|
|
|
|
|
// the names and orders are consistent with batchInfo
|
|
|
|
|
const int64_t cur_epoch_num_; // current epoch
|
|
|
|
|