Searched defs:TrainCallBackData (Results 1 – 1 of 1) sorted by relevance
31 struct TrainCallBackData { struct32 …TrainCallBackData(bool train_mode, int epoch, int step, Model *model): train_mode_(train_mode), ep… in TrainCallBackData() function55 virtual void Begin(const TrainCallBackData &cb_data) {} in Begin() argument