Class Config

tools/inference/java/src/main/java/Config.java:52–128

import org.apache.commons.cli.*;

public class Config {
    public static int threadNum;
    public static int batchSize;
    public static float cpuUsageRatio;
    public static int iteration;
    public static String outPerformanceFile;
	public static String inputdata;
	public static String modelFile;

    public Config() {};

    public static void ReadConfig(String[] args) {
        System.out.println(Arrays.asList(args));
		Option opt1 = new Option("t", "threadNum", true, "threrad num");
		opt1.setRequired(true);
		Option opt2 = new Option("bsz", "batchSize", true, "batch size");
		opt2.setRequired(true);
		Option opt3 = new Option("cr", "cpuRatio", true, "cpu usage ratio");
		opt3.setRequired(true);
        Option opt4 = new Option("it", "iteration", true, "iteration num");
        opt4.setRequired(true);
        Option opt5 = new Option("op", "outPerformanceFile", true, "perfomance file");
        opt5.setRequired(true);
		Option opt6 = new Option("inputdata", "inputdata", true, "training file");
        opt6.setRequired(true);
		Option opt7 = new Option("modelFile", "modelFile", true, "model file");
        opt7.setRequired(true);

		Options options = new Options();
		options.addOption(opt1);
		options.addOption(opt2);
		options.addOption(opt3);
        options.addOption(opt4);
        options.addOption(opt5);
		options.addOption(opt6);
		options.addOption(opt7);
        
		CommandLine cli = null;
		CommandLineParser cliParser = new DefaultParser();
		HelpFormatter helpFormatter = new HelpFormatter();
		try {
			cli = cliParser.parse(options, args);
		} catch (ParseException e) {
			helpFormatter.printHelp(">>>>>> test cli options", options);
			e.printStackTrace();
		}

		if (cli.hasOption("t")) {
			threadNum = Integer.parseInt(cli.getOptionValue("t", "1"));
			System.out.println(String.format(">>>>>> thread num: %s", threadNum));
		}
		if (cli.hasOption("bsz")) {
			batchSize = Integer.parseInt(cli.getOptionValue("bsz", "1"));
			System.out.println(String.format(">>>>>> batch size: %s", batchSize));
		}
        if (cli.hasOption("cr")) {
            cpuUsageRatio = Float.parseFloat(cli.getOptionValue("cr", "1.0"));
            System.out.println(String.format(">>>>>> cpu usage ratio: %s", cpuUsageRatio));