marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-08-14 16:50:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

add fallback option for sampling, for back-compat

Browse Source
This commit is contained in:
Marcin Junczys-Dowmunt 2022-05-09 13:28:28 -07:00
parent 1a74358277
commit e4f3d0f740
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/models/model_factory.cpp
View File

@ -374,7 +374,10 @@ Ptr<IModel> createModelFromOptions(Ptr<Options> options, usage use) {
auto sampling = options->get<std::vector<std::string>>("output-sampling", {});
std::string method = sampling.size() > 0 ? sampling[0] : "full";
if(method == "full" || method == "1" /*for backwards-compat when output-sampling: true in yaml file*/) {
if(method == "0") { /*for backwards-compat when output-sampling: false in yaml file*/
// do normal decoding
return New<Stepwise>(std::dynamic_pointer_cast<EncoderDecoder>(baseModel), New<LogSoftmaxStep>());
} else if(method == "full" || method == "1" /*for backwards-compat when output-sampling: true in yaml file*/) {
LOG(info, "Output sampling from the full softmax distribution");
return New<Stepwise>(std::dynamic_pointer_cast<EncoderDecoder>(baseModel), New<GumbelSoftmaxStep>());
} else if(method == "topk") {