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

cmd

package
v0.3.6 Latest Latest
Warning

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

Go to latest
Published: Feb 8, 2021 License: NCSA Imports: 13 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 NewProgress added in v0.3.0

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

func ParseModelName added in v0.2.19

func ParseModelName(model string) (string, 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