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