| 31 | namespace lbann { |
| 32 | |
| 33 | void construct_std_options() |
| 34 | { |
| 35 | auto& arg_parser = global_argument_parser(); |
| 36 | |
| 37 | // Bool flags |
| 38 | arg_parser.add_flag( |
| 39 | LBANN_OPTION_DISABLE_BACKGROUND_IO_ACTIVITY, |
| 40 | {"--disable_background_io_activity"}, |
| 41 | utils::ENV("LBANN_DISABLE_BACKGROUND_IO_ACTIVITY"), |
| 42 | "[STD] prevent the data coordinator from fetching data in the background"); |
| 43 | arg_parser.add_flag( |
| 44 | LBANN_OPTION_DISABLE_CUDA, |
| 45 | {"--disable_cuda"}, |
| 46 | "[STD] has no effect unless LBANN was compiled with LBANN_HAS_CUDNN"); |
| 47 | arg_parser.add_flag( |
| 48 | LBANN_OPTION_DISABLE_DISTCONV, |
| 49 | {"--disable_distconv"}, |
| 50 | utils::ENV("LBANN_DISABLE_DISTCONV"), |
| 51 | "[STD] Disables distconv support. Has no effect unless LBANN was compiled " |
| 52 | "with LBANN_HAS_DISTCONV"); |
| 53 | arg_parser.add_flag( |
| 54 | LBANN_OPTION_DISABLE_SIGNAL_HANDLER, |
| 55 | {"--disable_signal_handler"}, |
| 56 | "[STD] Disables signal handling (signal handling on by default)"); |
| 57 | arg_parser.add_flag(LBANN_OPTION_EXIT_AFTER_SETUP, |
| 58 | {"--exit_after_setup"}, |
| 59 | "[STD] Forces exit after model setup"); |
| 60 | arg_parser.add_flag(LBANN_OPTION_GENERATE_MULTI_PROTO, |
| 61 | {"--generate_multi_proto"}, |
| 62 | "[STD] Enables loading of multiple prototext files for " |
| 63 | "model, datareader, optimizer, etc. input options"); |
| 64 | arg_parser.add_flag( |
| 65 | LBANN_OPTION_LOAD_MODEL_WEIGHTS_DIR_IS_COMPLETE, |
| 66 | {"--load_model_weights_dir_is_complete"}, |
| 67 | "[STD] Use load_model_weights_dir as given, ignoring checkpoint hierarchy"); |
| 68 | arg_parser.add_flag( |
| 69 | LBANN_OPTION_LTFB_ALLOW_GLOBAL_STATISTICS, |
| 70 | {"--ltfb_allow_global_statistics"}, |
| 71 | utils::ENV("LBANN_LTFB_ALLOW_GLOBAL_STATISTICS"), |
| 72 | "[STD, deprecated] Allow the print_statistics callback to report " |
| 73 | "global (inter-trainer) summary statistics."); |
| 74 | arg_parser.add_flag(LBANN_OPTION_LTFB_VERBOSE, |
| 75 | {"--ltfb_verbose"}, |
| 76 | utils::ENV("LBANN_LTFB_VERBOSE"), |
| 77 | "[STD, deprecated] Increases number of per-trainer " |
| 78 | "messages that are reported"); |
| 79 | arg_parser.add_flag(LBANN_OPTION_ALLOW_MULTITRAINER_GLOBAL_STATISTICS, |
| 80 | {"--ltfb_global_multitrainer_statistics"}, |
| 81 | utils::ENV("LBANN_ALLOW_MULTITRAINER_GLOBAL_STATISTICS"), |
| 82 | "[STD] Allow the print_statistics callback to report " |
| 83 | "global (inter-trainer) summary statistics."); |
| 84 | arg_parser.add_flag( |
| 85 | LBANN_OPTION_MULTITRAINER_VERBOSE, |
| 86 | {"--multitrainer_verbose"}, |
| 87 | utils::ENV("LBANN_MULTITRAINER_VERBOSE"), |
| 88 | "[STD] Increases number of per-trainer messages that are reported"); |
| 89 | arg_parser.add_flag(LBANN_OPTION_PRELOAD_DATA_STORE, |
| 90 | {"--preload_data_store"}, |