Help us shape the future of Go! Take the 2026 Go Developer Survey

cmd

package
v0.2.19 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Jan 9, 2018 License: NCSA Imports: 11 Imported by: 2

Documentation

Index

Constants

This section is empty.

Variables

View Source
var (
	IsDebug   bool
	IsVerbose bool
	AppSecret string
	CfgFile   string
)
View Source
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 added in v0.2.19

func Info()

func Init

func Init()

Init reads in config file and ENV variables if set.

func NVIDIASmi added in v0.2.19

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

func ParseModelName added in v0.2.19

func ParseModelName(model string) (modelName, version string)

Types

This section is empty.

Directories

Path Synopsis
run command

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL