predictGeneral

package
v1.4.21 Latest Latest
Warning

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

Go to latest
Published: Jun 14, 2022 License: NCSA Imports: 35 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func InitCorePrediction

func InitCorePrediction() (err error)

Types

type InferenceResponse added in v1.4.20

type InferenceResponse struct {
	InferenceDuration   time.Duration
	PostprocessDuration time.Duration
	PreprocessDuration  time.Duration
	Responses           []*dl.FeatureResponse
	TotalDuration       time.Duration
	WarmupDuration      time.Duration
}

func CorePrediction

func CorePrediction(model *dl.ModelManifest, predictors []common.Predictor, settings Settings) (inferenceResponse *InferenceResponse, err error)

type Settings

type Settings struct {
	BatchSize                  int
	UseGPU                     bool
	GpuDeviceId                int
	GpuMetrics                 string
	PublishPredictions         bool
	PublishToDatabase          bool
	PartitionListSize          int
	SaveInferenceResult        bool
	FailOnFirstError           bool
	DisableFrameworkAutoTuning bool
	DesiredResultModality      string

	TracerServerAddr string
	TracePreprocess  bool
	TraceLevelName   string
	TraceLevel       tracer.Level
	TracerAddress    string

	DatabaseAddress      string
	DatabaseName         string
	DatabaseEndpoints    []string
	DefaultChannelBuffer int

	BaseDir          string
	TimeoutOptionSet bool
	DuplicateInput   int
	NumWarmUp        int

	DatasetPath      string
	DatasetImageUrls []string
	Count            int
}

func GetDefaultSettings

func GetDefaultSettings() (o Settings)

Jump to

Keyboard shortcuts

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