diff options
Diffstat (limited to 'src/common/aliases.cpp')
-rw-r--r-- | src/common/aliases.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/common/aliases.cpp b/src/common/aliases.cpp index 0be26a8c..99574fe1 100644 --- a/src/common/aliases.cpp +++ b/src/common/aliases.cpp @@ -29,8 +29,8 @@ void ConfigParser::addAliases(cli::CLIWrapper& cli) { cli.alias("fp16", "true", [&](YAML::Node& config) { if(mode_ == cli::mode::training) { config["precision"] = std::vector<std::string>({"float16", "float32"}); // inference type, optimization type, save type - // scaling factor (power of 2), frequency, multiplier at increase, tolerance, range, minium factor - config["cost-scaling"] = std::vector<std::string>({"0", "1000", "2", "0.05", "10", "1e-5"}); + // scaling factor, frequency, multiplier at increase, minium scaling factor + config["cost-scaling"] = std::vector<std::string>({"256.f", "1000", "2.f", "256.f"}); } else { config["precision"] = std::vector<std::string>({"float16"}); // for inference we do not need the other types } |