cmd

package
v1.4.17 Latest Latest
Warning

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

Go to latest
Published: Apr 18, 2022 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{
		"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 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