cmd

package
v1.1.1 Latest Latest
Warning

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

Go to latest
Published: Mar 20, 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

Jump to

Keyboard shortcuts

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