com.intel.analytics.bigdl.example.mkldnn.int8.Utils.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bigdl-SPARK_3.1 Show documentation
The newest version!
/*
 * Copyright 2016 The BigDL Authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.intel.analytics.bigdl.example.mkldnn.int8

import scopt.OptionParser

object Utils {
  case class TestParams(
    folder: String = "./",
    model: String = "",
    batchSize: Int = 128
  )

  val testParser: OptionParser[TestParams] = new OptionParser[TestParams](
    "BigDL ResNet-50 with mkldnn int8 on ImageNet Test Example") {
    opt[String]('f', "folder")
      .text("the location of imagenet dataset")
      .action((x, c) => c.copy(folder = x))
    opt[String]('m', "model")
      .text("the location of model snapshot")
      .action((x, c) => c.copy(model = x))
      .required()
      .required()
    opt[Int]('b', "batchSize")
      .text("batch size")
      .action((x, c) => c.copy(batchSize = x))
  }

  case class GenInt8ScalesParams(
    folder: String = "./",
    model: String = "",
    batchSize: Int = 128,
    numOfBatch: Int = 1
  )

  val genInt8ScalesParser: OptionParser[GenInt8ScalesParams] =
    new OptionParser[GenInt8ScalesParams](
      "BigDL ResNet-50 generate scales on ImageNet Test Example") {
      opt[String]('f', "folder")
        .text("the location of imagenet dataset")
        .action((x, c) => c.copy(folder = x))
      opt[String]('m', "model")
        .text("the location of model snapshot")
        .action((x, c) => c.copy(model = x))
        .required()
        .required()
      opt[Int]('b', "batchSize")
        .text("batch size")
        .action((x, c) => c.copy(batchSize = x))
      opt[Int]('n', "numOfBatch")
        .action((x, c) => c.copy(batchSize = x))
    }
}