Function construct_opts

core-driver/main.cpp:32–62

#include <stdio.h>

void construct_opts(int argc, char **argv) {
  auto& arg_parser = lbann::global_argument_parser();
  arg_parser.add_option("samples",
                        {"-n"},
                        "Number of samples to run inference on",
                        128);
  arg_parser.add_option("channels",
                        {"-c"},
                        "Number of image channels in sample",
                        1);
  arg_parser.add_option("height",
                        {"-h"},
                        "Height of image in sample",
                        28);
  arg_parser.add_option("width",
                        {"-w"},
                        "Width of image in sample",
                        28);
  arg_parser.add_option("labels",
                        {"-l"},
                        "Number of labels in dataset",
                        10);
  arg_parser.add_option("minibatchsize",
                        {"-mbs"},
                        "Number of samples in a mini-batch",
                        16);
  arg_parser.add_required_argument<std::string>
                                  ("model",
                                   "Directory containing checkpointed model");
  arg_parser.parse(argc, argv);
}

El::DistMatrix<float, El::STAR, El::STAR, El::ELEMENT, El::Device::CPU>
random_samples(El::Grid const& g, int n, int c, int h, int w) {