cmd

Documentation

Index

• Variables
• func Init()
• func NewProgress(prefix string, count int) *pb.ProgressBar
• func ParseModelName(model string) (string, string)

Variables

var (
	IsDebug   bool
	IsVerbose bool
	AppSecret string
	CfgFile   string
)

var (
	DefaultEvaulationModels = []string{
		"TorchVision_AlexNet_1.0",
	}

	DefaultEvaluationFrameworks = []string{
		"pytorch",
	}

	FrameworkNames = map[string]string{
		"tensorflow":  "TensorFlow",
		"tensorrt":    "TensorRT",
		"mxnet":       "MXNet",
		"caffe":       "Caffe",
		"caffe2":      "Caffe2",
		"cntk":        "CNTK",
		"pytorch":     "PyTorch",
		"onnxruntime": "ONNXRuntime",
	}
)

Functions

func Init

func Init()

Init reads in config file and ENV variables if set.

func NewProgress

func NewProgress(prefix string, count int) *pb.ProgressBar

func ParseModelName

func ParseModelName(model string) (string, string)