File tree Expand file tree Collapse file tree 1 file changed +17
-2
lines changed
Expand file tree Collapse file tree 1 file changed +17
-2
lines changed Original file line number Diff line number Diff line change @@ -230,6 +230,21 @@ int example_regularstepgradientdescent( int argc, char* argv[] )
230230
231231int main ( int argc, char * argv[] )
232232{
233- return example_gradientdescent ( argc, argv );
234- // return example_regularstepgradientdescent( argc, argv );
233+ if ( argc < 3 )
234+ {
235+ std::cerr << " Usage: MinimalPathExamples <SpeedImage> <OutputImage> [UseRegularStepGradientDescentOptimizer]" << std::endl;
236+ return EXIT_FAILURE;
237+ }
238+
239+ int useRegularStepGradientDescentOptimizer = 0 ;
240+ if ( argc > 3 )
241+ {
242+ useRegularStepGradientDescentOptimizer = std::atoi ( argv[3 ] );
243+ }
244+
245+ if ( useRegularStepGradientDescentOptimizer )
246+ {
247+ return example_regularstepgradientdescent ( argc, argv );
248+ }
249+ return example_gradientdescent ( argc, argv );
235250}
You can’t perform that action at this time.
0 commit comments