cmd

Documentation

Index

• Variables
• func Info()
• func Init()
• func NVIDIASmi() (*nvidiasmi.NvidiaSmi, error)
• 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{
		"SqueezeNet_1.0",
		"SqueezeNet_1.1",
		"BVLC_AlexNet_1.0",
		"BVLC_Reference_CaffeNet_1.0",
		"BVLC_GoogLeNet_1.0",
		"ResNet101_1.0",
		"ResNet101_2.0",
		"WRN50_2.0",
		"BVLC_Reference_RCNN_ILSVRC13_1.0",
		"Inception_3.0",
		"Inception_4.0",
		"ResNeXt50_32x4d_1.0",
		"VGG16_1.0",
		"VGG19_1.0",
	}

	DefaultEvaluationFrameworks = []string{
		"mxnet",
		"cntk",
		"caffe2",
		"tensorflow",
		"tensorrt",
		"caffe",
	}

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

Functions

func Info

func Info()

func Init

func Init()

Init reads in config file and ENV variables if set.

func NVIDIASmi

func NVIDIASmi() (*nvidiasmi.NvidiaSmi, error)

func NewProgress

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

func ParseModelName

func ParseModelName(model string) (string, string)