marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-09-17 09:47:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix fallback to default paths in MNIST example (#821)

Browse Source 
If --train-sets or --valid-sets are not provided, the fallback to the
hard-coded paths does not occur. This commit requires that these
entities have a non-empty value.
This commit is contained in:
Graeme 2021-03-12 09:42:43 +00:00 committed by GitHub
parent f74d055d20
commit db399d749c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/examples/mnist/mnist_ffnn.cpp
View File

@ -24,9 +24,9 @@ using namespace marian;
int main(int argc, char** argv) {
auto options = parseOptions(argc, argv, cli::mode::training, false);
if(!options->has("train-sets"))
if(!options->hasAndNotEmpty("train-sets"))
options->set("train-sets", TRAIN_SET);
if(!options->has("valid-sets"))
if(!options->hasAndNotEmpty("valid-sets"))
options->set("valid-sets", VALID_SET);
if(options->get<std::string>("type") != "mnist-lenet")