Class TrainCfg
Defined in File cfg.h
Class Documentation
-
class TrainCfg
Public Members
-
std::string loss_name_
Set part of the name that identify a loss kernel
-
MixPrecisionCfg mix_precision_cfg_
Mix precision configuration
-
std::string loss_name_