diff --git a/waifu2x-caffe/Source.cpp b/waifu2x-caffe/Source.cpp index f28d002..e32b407 100644 --- a/waifu2x-caffe/Source.cpp +++ b/waifu2x-caffe/Source.cpp @@ -89,7 +89,7 @@ int main(int argc, char** argv) TCLAP::ValueArg cmdModelPath("", "model_dir", "path to custom model directory (don't append last / )", false, - "models", "string", cmd); + "models/anime_style_art_rgb", "string", cmd); std::vector cmdProcessConstraintV; cmdProcessConstraintV.push_back("cpu");