Documentation ¶
Index ¶
- type Job
- type ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Annotations(annotations map[string]string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Build() (*Job, error)
- func (m *ModelBenchmarkArgsBuilder) CPU(cpu string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Command(args []string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Concurrency(concurrency int) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) DataDirs(volumes map[string]string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Datas(volumes map[string]string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Duration(duration int) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Envs(envs map[string]string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) GPUCore(core int) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) GPUCount(count int) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) GPUMemory(memory int) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Image(image string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) ImagePullPolicy(policy string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) ImagePullSecrets(secrets []string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Inputs(inputs string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Labels(labels map[string]string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Memory(memory string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) ModelConfigFile(filePath string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) ModelName(name string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) ModelPath(path string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Name(name string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Namespace(namespace string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) NodeSelectors(selectors map[string]string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Outputs(outputs string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) ReportPath(reportPath string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Requests(requests int) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Shell(shell string) *ModelBenchmarkArgsBuilder
- func (m *ModelBenchmarkArgsBuilder) Tolerations(tolerations []string) *ModelBenchmarkArgsBuilder
- type ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Annotations(annotations map[string]string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) BatchSize(batchSize int) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Build() (*Job, error)
- func (m *ModelEvaluateJobBuilder) CPU(cpu string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Command(args []string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) DataDirs(volumes map[string]string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Datas(volumes map[string]string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) DatasetPath(datasetPath string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Envs(envs map[string]string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) GPUCore(core int) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) GPUCount(count int) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) GPUMemory(memory int) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Image(image string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) ImagePullPolicy(policy string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) ImagePullSecrets(secrets []string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Labels(labels map[string]string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Memory(memory string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) ModelPath(path string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) ModelPlatform(modelPlatform string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Name(name string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Namespace(namespace string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) NodeSelectors(selectors map[string]string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) ReportPath(reportPath string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Shell(shell string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) SyncImage(image string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) SyncMode(mode string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) SyncSource(source string) *ModelEvaluateJobBuilder
- func (m *ModelEvaluateJobBuilder) Tolerations(tolerations []string) *ModelEvaluateJobBuilder
- type ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Annotations(annotations map[string]string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Build() (*Job, error)
- func (m *ModelOptimizeJobBuilder) CPU(cpu string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Command(args []string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) DataDirs(volumes map[string]string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Datas(volumes map[string]string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Envs(envs map[string]string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) ExportPath(exportPath string) *ModelOptimizeJobBuilder
- func (b *ModelOptimizeJobBuilder) GPUCore(core int) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) GPUCount(count int) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) GPUMemory(memory int) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Image(image string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) ImagePullPolicy(policy string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) ImagePullSecrets(secrets []string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Inputs(inputs string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Labels(labels map[string]string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Memory(memory string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) ModelConfigFile(filePath string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) ModelName(name string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) ModelPath(path string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Name(name string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Namespace(namespace string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) NodeSelectors(selectors map[string]string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Optimizer(optimizer string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Outputs(outputs string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Shell(shell string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) TargetDevice(targetDevice string) *ModelOptimizeJobBuilder
- func (m *ModelOptimizeJobBuilder) Tolerations(tolerations []string) *ModelOptimizeJobBuilder
- type ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Annotations(annotations map[string]string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Build() (*Job, error)
- func (m *ModelProfileJobBuilder) CPU(cpu string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Command(args []string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) DataDirs(volumes map[string]string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Datas(volumes map[string]string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Envs(envs map[string]string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) GPUCore(core int) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) GPUCount(count int) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) GPUMemory(memory int) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Image(image string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) ImagePullPolicy(policy string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) ImagePullSecrets(secrets []string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Inputs(inputs string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Labels(labels map[string]string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Memory(memory string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) ModelConfigFile(filePath string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) ModelName(name string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) ModelPath(path string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Name(name string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Namespace(namespace string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) NodeSelectors(selectors map[string]string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Outputs(outputs string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) ReportPath(reportPath string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Shell(shell string) *ModelProfileJobBuilder
- func (m *ModelProfileJobBuilder) Tolerations(tolerations []string) *ModelProfileJobBuilder
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Job ¶
type Job struct {
// contains filtered or unexported fields
}
Job defines the base job
func (*Job) Type ¶
func (j *Job) Type() types.ModelJobType
type ModelBenchmarkArgsBuilder ¶
type ModelBenchmarkArgsBuilder struct { argsbuilder.ArgsBuilder // contains filtered or unexported fields }
func NewModelBenchmarkArgsBuilder ¶
func NewModelBenchmarkArgsBuilder() *ModelBenchmarkArgsBuilder
func (*ModelBenchmarkArgsBuilder) Annotations ¶
func (m *ModelBenchmarkArgsBuilder) Annotations(annotations map[string]string) *ModelBenchmarkArgsBuilder
Annotations is used to add annotations for job pods,match option --annotation
func (*ModelBenchmarkArgsBuilder) Build ¶
func (m *ModelBenchmarkArgsBuilder) Build() (*Job, error)
Build is used to build the job
func (*ModelBenchmarkArgsBuilder) CPU ¶
func (m *ModelBenchmarkArgsBuilder) CPU(cpu string) *ModelBenchmarkArgsBuilder
CPU assign cpu limits,match the option --cpu
func (*ModelBenchmarkArgsBuilder) Command ¶
func (m *ModelBenchmarkArgsBuilder) Command(args []string) *ModelBenchmarkArgsBuilder
Command is used to set job command
func (*ModelBenchmarkArgsBuilder) Concurrency ¶
func (m *ModelBenchmarkArgsBuilder) Concurrency(concurrency int) *ModelBenchmarkArgsBuilder
Concurrency is used to specify number of concurrently to run
func (*ModelBenchmarkArgsBuilder) DataDirs ¶
func (m *ModelBenchmarkArgsBuilder) DataDirs(volumes map[string]string) *ModelBenchmarkArgsBuilder
DataDirs is used to mount host files to job containers,match option --data-dir
func (*ModelBenchmarkArgsBuilder) Datas ¶
func (m *ModelBenchmarkArgsBuilder) Datas(volumes map[string]string) *ModelBenchmarkArgsBuilder
Datas is used to mount k8s pvc to job pods,match option --data
func (*ModelBenchmarkArgsBuilder) Duration ¶
func (m *ModelBenchmarkArgsBuilder) Duration(duration int) *ModelBenchmarkArgsBuilder
Duration is used to specify benchmark duration
func (*ModelBenchmarkArgsBuilder) Envs ¶
func (m *ModelBenchmarkArgsBuilder) Envs(envs map[string]string) *ModelBenchmarkArgsBuilder
Envs is used to set env of job containers,match option --env
func (*ModelBenchmarkArgsBuilder) GPUCore ¶
func (m *ModelBenchmarkArgsBuilder) GPUCore(core int) *ModelBenchmarkArgsBuilder
GPUCore is used to set gpu core for the job,match the option --gpucore
func (*ModelBenchmarkArgsBuilder) GPUCount ¶
func (m *ModelBenchmarkArgsBuilder) GPUCount(count int) *ModelBenchmarkArgsBuilder
GPUCount is used to set count of gpu for the job,match the option --gpus
func (*ModelBenchmarkArgsBuilder) GPUMemory ¶
func (m *ModelBenchmarkArgsBuilder) GPUMemory(memory int) *ModelBenchmarkArgsBuilder
GPUMemory is used to set gpu memory for the job,match the option --gpumemory
func (*ModelBenchmarkArgsBuilder) Image ¶
func (m *ModelBenchmarkArgsBuilder) Image(image string) *ModelBenchmarkArgsBuilder
Image is used to set job image,match the option --image
func (*ModelBenchmarkArgsBuilder) ImagePullPolicy ¶
func (m *ModelBenchmarkArgsBuilder) ImagePullPolicy(policy string) *ModelBenchmarkArgsBuilder
ImagePullPolicy is used to set image pull policy,match the option --image-pull-policy
func (*ModelBenchmarkArgsBuilder) ImagePullSecrets ¶
func (m *ModelBenchmarkArgsBuilder) ImagePullSecrets(secrets []string) *ModelBenchmarkArgsBuilder
ImagePullSecrets is used to set image pull secrests,match option --image-pull-secret
func (*ModelBenchmarkArgsBuilder) Inputs ¶
func (m *ModelBenchmarkArgsBuilder) Inputs(inputs string) *ModelBenchmarkArgsBuilder
Inputs is used to specify model inputs
func (*ModelBenchmarkArgsBuilder) Labels ¶
func (m *ModelBenchmarkArgsBuilder) Labels(labels map[string]string) *ModelBenchmarkArgsBuilder
Labels is used to add labels for job
func (*ModelBenchmarkArgsBuilder) Memory ¶
func (m *ModelBenchmarkArgsBuilder) Memory(memory string) *ModelBenchmarkArgsBuilder
Memory assign memory limits,match option --memory
func (*ModelBenchmarkArgsBuilder) ModelConfigFile ¶
func (m *ModelBenchmarkArgsBuilder) ModelConfigFile(filePath string) *ModelBenchmarkArgsBuilder
ModelConfigFile is used to set model config file,match the option --model-config-file
func (*ModelBenchmarkArgsBuilder) ModelName ¶
func (m *ModelBenchmarkArgsBuilder) ModelName(name string) *ModelBenchmarkArgsBuilder
ModelName is used to set model name,match the option --model-name
func (*ModelBenchmarkArgsBuilder) ModelPath ¶
func (m *ModelBenchmarkArgsBuilder) ModelPath(path string) *ModelBenchmarkArgsBuilder
ModelPath is used to set model path,match the option --model-path
func (*ModelBenchmarkArgsBuilder) Name ¶
func (m *ModelBenchmarkArgsBuilder) Name(name string) *ModelBenchmarkArgsBuilder
Name is used to set job name,match option --name
func (*ModelBenchmarkArgsBuilder) Namespace ¶
func (m *ModelBenchmarkArgsBuilder) Namespace(namespace string) *ModelBenchmarkArgsBuilder
Namespace is used to set job namespace,match option --namespace
func (*ModelBenchmarkArgsBuilder) NodeSelectors ¶
func (m *ModelBenchmarkArgsBuilder) NodeSelectors(selectors map[string]string) *ModelBenchmarkArgsBuilder
NodeSelectors is used to set node selectors for scheduling job,match option --selector
func (*ModelBenchmarkArgsBuilder) Outputs ¶
func (m *ModelBenchmarkArgsBuilder) Outputs(outputs string) *ModelBenchmarkArgsBuilder
Outputs is used to specify model outputs
func (*ModelBenchmarkArgsBuilder) ReportPath ¶
func (m *ModelBenchmarkArgsBuilder) ReportPath(reportPath string) *ModelBenchmarkArgsBuilder
ReportPath is used to specify benchmark result saved path
func (*ModelBenchmarkArgsBuilder) Requests ¶
func (m *ModelBenchmarkArgsBuilder) Requests(requests int) *ModelBenchmarkArgsBuilder
Requests is used to specify number of requests to run
func (*ModelBenchmarkArgsBuilder) Shell ¶
func (m *ModelBenchmarkArgsBuilder) Shell(shell string) *ModelBenchmarkArgsBuilder
Shell is used to specify linux shell type
func (*ModelBenchmarkArgsBuilder) Tolerations ¶
func (m *ModelBenchmarkArgsBuilder) Tolerations(tolerations []string) *ModelBenchmarkArgsBuilder
Tolerations is used to set tolerations for tolerate nodes,match option --toleration
type ModelEvaluateJobBuilder ¶
type ModelEvaluateJobBuilder struct { argsbuilder.ArgsBuilder // contains filtered or unexported fields }
func NewModelEvaluateJobBuilder ¶
func NewModelEvaluateJobBuilder() *ModelEvaluateJobBuilder
func (*ModelEvaluateJobBuilder) Annotations ¶
func (m *ModelEvaluateJobBuilder) Annotations(annotations map[string]string) *ModelEvaluateJobBuilder
Annotations is used to add annotations for job pods,match option --annotation
func (*ModelEvaluateJobBuilder) BatchSize ¶
func (m *ModelEvaluateJobBuilder) BatchSize(batchSize int) *ModelEvaluateJobBuilder
BatchSize is the batch size of evaluate
func (*ModelEvaluateJobBuilder) Build ¶
func (m *ModelEvaluateJobBuilder) Build() (*Job, error)
Build is used to build the job
func (*ModelEvaluateJobBuilder) CPU ¶
func (m *ModelEvaluateJobBuilder) CPU(cpu string) *ModelEvaluateJobBuilder
CPU assign cpu limits,match the option --cpu
func (*ModelEvaluateJobBuilder) Command ¶
func (m *ModelEvaluateJobBuilder) Command(args []string) *ModelEvaluateJobBuilder
Command is used to set job command
func (*ModelEvaluateJobBuilder) DataDirs ¶
func (m *ModelEvaluateJobBuilder) DataDirs(volumes map[string]string) *ModelEvaluateJobBuilder
DataDirs is used to mount host files to job containers,match option --data-dir
func (*ModelEvaluateJobBuilder) Datas ¶
func (m *ModelEvaluateJobBuilder) Datas(volumes map[string]string) *ModelEvaluateJobBuilder
Datas is used to mount k8s pvc to job pods,match option --data
func (*ModelEvaluateJobBuilder) DatasetPath ¶
func (m *ModelEvaluateJobBuilder) DatasetPath(datasetPath string) *ModelEvaluateJobBuilder
DatasetPath is the dataset to evaluate model
func (*ModelEvaluateJobBuilder) Envs ¶
func (m *ModelEvaluateJobBuilder) Envs(envs map[string]string) *ModelEvaluateJobBuilder
Envs is used to set env of job containers,match option --env
func (*ModelEvaluateJobBuilder) GPUCore ¶
func (m *ModelEvaluateJobBuilder) GPUCore(core int) *ModelEvaluateJobBuilder
GPUCore is used to set gpu core for the job,match the option --gpumemory
func (*ModelEvaluateJobBuilder) GPUCount ¶
func (m *ModelEvaluateJobBuilder) GPUCount(count int) *ModelEvaluateJobBuilder
GPUCount is used to set count of gpu for the job,match the option --gpus
func (*ModelEvaluateJobBuilder) GPUMemory ¶
func (m *ModelEvaluateJobBuilder) GPUMemory(memory int) *ModelEvaluateJobBuilder
GPUMemory is used to set gpu memory for the job,match the option --gpumemory
func (*ModelEvaluateJobBuilder) Image ¶
func (m *ModelEvaluateJobBuilder) Image(image string) *ModelEvaluateJobBuilder
Image is used to set job image,match the option --image
func (*ModelEvaluateJobBuilder) ImagePullPolicy ¶
func (m *ModelEvaluateJobBuilder) ImagePullPolicy(policy string) *ModelEvaluateJobBuilder
ImagePullPolicy is used to set image pull policy,match the option --image-pull-policy
func (*ModelEvaluateJobBuilder) ImagePullSecrets ¶
func (m *ModelEvaluateJobBuilder) ImagePullSecrets(secrets []string) *ModelEvaluateJobBuilder
ImagePullSecrets is used to set image pull secrests,match option --image-pull-secret
func (*ModelEvaluateJobBuilder) Labels ¶
func (m *ModelEvaluateJobBuilder) Labels(labels map[string]string) *ModelEvaluateJobBuilder
Labels is used to add labels for job
func (*ModelEvaluateJobBuilder) Memory ¶
func (m *ModelEvaluateJobBuilder) Memory(memory string) *ModelEvaluateJobBuilder
Memory assign memory limits,match option --memory
func (*ModelEvaluateJobBuilder) ModelPath ¶
func (m *ModelEvaluateJobBuilder) ModelPath(path string) *ModelEvaluateJobBuilder
ModelPath is used to set model path,match the option --model-path
func (*ModelEvaluateJobBuilder) ModelPlatform ¶
func (m *ModelEvaluateJobBuilder) ModelPlatform(modelPlatform string) *ModelEvaluateJobBuilder
ModelPlatform specify the model platform, such as torchscript/tensorflow
func (*ModelEvaluateJobBuilder) Name ¶
func (m *ModelEvaluateJobBuilder) Name(name string) *ModelEvaluateJobBuilder
Name is used to set job name,match option --name
func (*ModelEvaluateJobBuilder) Namespace ¶
func (m *ModelEvaluateJobBuilder) Namespace(namespace string) *ModelEvaluateJobBuilder
Namespace is used to set job namespace,match option --namespace
func (*ModelEvaluateJobBuilder) NodeSelectors ¶
func (m *ModelEvaluateJobBuilder) NodeSelectors(selectors map[string]string) *ModelEvaluateJobBuilder
NodeSelectors is used to set node selectors for scheduling job,match option --selector
func (*ModelEvaluateJobBuilder) ReportPath ¶
func (m *ModelEvaluateJobBuilder) ReportPath(reportPath string) *ModelEvaluateJobBuilder
ReportPath is used to specify evaluate result path
func (*ModelEvaluateJobBuilder) Shell ¶
func (m *ModelEvaluateJobBuilder) Shell(shell string) *ModelEvaluateJobBuilder
Shell is used to specify linux shell type
func (*ModelEvaluateJobBuilder) SyncImage ¶
func (m *ModelEvaluateJobBuilder) SyncImage(image string) *ModelEvaluateJobBuilder
SyncImage is used to set syncing image,match option --sync-image
func (*ModelEvaluateJobBuilder) SyncMode ¶
func (m *ModelEvaluateJobBuilder) SyncMode(mode string) *ModelEvaluateJobBuilder
SyncMode is used to set syncing mode,match option --sync-mode
func (*ModelEvaluateJobBuilder) SyncSource ¶
func (m *ModelEvaluateJobBuilder) SyncSource(source string) *ModelEvaluateJobBuilder
SyncSource is used to set syncing source,match option --sync-source
func (*ModelEvaluateJobBuilder) Tolerations ¶
func (m *ModelEvaluateJobBuilder) Tolerations(tolerations []string) *ModelEvaluateJobBuilder
Tolerations is used to set tolerations for tolerate nodes,match option --toleration
type ModelOptimizeJobBuilder ¶
type ModelOptimizeJobBuilder struct { argsbuilder.ArgsBuilder // contains filtered or unexported fields }
func NewModelOptimizeJobBuilder ¶
func NewModelOptimizeJobBuilder() *ModelOptimizeJobBuilder
func (*ModelOptimizeJobBuilder) Annotations ¶
func (m *ModelOptimizeJobBuilder) Annotations(annotations map[string]string) *ModelOptimizeJobBuilder
Annotations is used to add annotations for job pods,match option --annotation
func (*ModelOptimizeJobBuilder) Build ¶
func (m *ModelOptimizeJobBuilder) Build() (*Job, error)
Build is used to build the job
func (*ModelOptimizeJobBuilder) CPU ¶
func (m *ModelOptimizeJobBuilder) CPU(cpu string) *ModelOptimizeJobBuilder
CPU assign cpu limits,match the option --cpu
func (*ModelOptimizeJobBuilder) Command ¶
func (m *ModelOptimizeJobBuilder) Command(args []string) *ModelOptimizeJobBuilder
Command is used to set job command
func (*ModelOptimizeJobBuilder) DataDirs ¶
func (m *ModelOptimizeJobBuilder) DataDirs(volumes map[string]string) *ModelOptimizeJobBuilder
DataDirs is used to mount host files to job containers,match option --data-dir
func (*ModelOptimizeJobBuilder) Datas ¶
func (m *ModelOptimizeJobBuilder) Datas(volumes map[string]string) *ModelOptimizeJobBuilder
Datas is used to mount k8s pvc to job pods,match option --data
func (*ModelOptimizeJobBuilder) Envs ¶
func (m *ModelOptimizeJobBuilder) Envs(envs map[string]string) *ModelOptimizeJobBuilder
Envs is used to set env of job containers,match option --env
func (*ModelOptimizeJobBuilder) ExportPath ¶
func (m *ModelOptimizeJobBuilder) ExportPath(exportPath string) *ModelOptimizeJobBuilder
ExportPath is used to specify optimized model save path
func (*ModelOptimizeJobBuilder) GPUCore ¶
func (b *ModelOptimizeJobBuilder) GPUCore(core int) *ModelOptimizeJobBuilder
GPUCore is used to set gpu core for the job, match the option --gpucore
func (*ModelOptimizeJobBuilder) GPUCount ¶
func (m *ModelOptimizeJobBuilder) GPUCount(count int) *ModelOptimizeJobBuilder
GPUCount is used to set count of gpu for the job,match the option --gpus
func (*ModelOptimizeJobBuilder) GPUMemory ¶
func (m *ModelOptimizeJobBuilder) GPUMemory(memory int) *ModelOptimizeJobBuilder
GPUMemory is used to set gpu memory for the job,match the option --gpumemory
func (*ModelOptimizeJobBuilder) Image ¶
func (m *ModelOptimizeJobBuilder) Image(image string) *ModelOptimizeJobBuilder
Image is used to set job image,match the option --image
func (*ModelOptimizeJobBuilder) ImagePullPolicy ¶
func (m *ModelOptimizeJobBuilder) ImagePullPolicy(policy string) *ModelOptimizeJobBuilder
ImagePullPolicy is used to set image pull policy,match the option --image-pull-policy
func (*ModelOptimizeJobBuilder) ImagePullSecrets ¶
func (m *ModelOptimizeJobBuilder) ImagePullSecrets(secrets []string) *ModelOptimizeJobBuilder
ImagePullSecrets is used to set image pull secrests,match option --image-pull-secret
func (*ModelOptimizeJobBuilder) Inputs ¶
func (m *ModelOptimizeJobBuilder) Inputs(inputs string) *ModelOptimizeJobBuilder
Inputs is used to specify model inputs
func (*ModelOptimizeJobBuilder) Labels ¶
func (m *ModelOptimizeJobBuilder) Labels(labels map[string]string) *ModelOptimizeJobBuilder
Labels is used to add labels for job
func (*ModelOptimizeJobBuilder) Memory ¶
func (m *ModelOptimizeJobBuilder) Memory(memory string) *ModelOptimizeJobBuilder
Memory assign memory limits,match option --memory
func (*ModelOptimizeJobBuilder) ModelConfigFile ¶
func (m *ModelOptimizeJobBuilder) ModelConfigFile(filePath string) *ModelOptimizeJobBuilder
ModelConfigFile is used to set model config file,match the option --model-config-file
func (*ModelOptimizeJobBuilder) ModelName ¶
func (m *ModelOptimizeJobBuilder) ModelName(name string) *ModelOptimizeJobBuilder
ModelName is used to set model name,match the option --model-name
func (*ModelOptimizeJobBuilder) ModelPath ¶
func (m *ModelOptimizeJobBuilder) ModelPath(path string) *ModelOptimizeJobBuilder
ModelPath is used to set model path,match the option --model-path
func (*ModelOptimizeJobBuilder) Name ¶
func (m *ModelOptimizeJobBuilder) Name(name string) *ModelOptimizeJobBuilder
Name is used to set job name,match option --name
func (*ModelOptimizeJobBuilder) Namespace ¶
func (m *ModelOptimizeJobBuilder) Namespace(namespace string) *ModelOptimizeJobBuilder
Namespace is used to set job namespace,match option --namespace
func (*ModelOptimizeJobBuilder) NodeSelectors ¶
func (m *ModelOptimizeJobBuilder) NodeSelectors(selectors map[string]string) *ModelOptimizeJobBuilder
NodeSelectors is used to set node selectors for scheduling job,match option --selector
func (*ModelOptimizeJobBuilder) Optimizer ¶
func (m *ModelOptimizeJobBuilder) Optimizer(optimizer string) *ModelOptimizeJobBuilder
Optimizer is used to specify optimized model save path
func (*ModelOptimizeJobBuilder) Outputs ¶
func (m *ModelOptimizeJobBuilder) Outputs(outputs string) *ModelOptimizeJobBuilder
Outputs is used to specify model outputs
func (*ModelOptimizeJobBuilder) Shell ¶
func (m *ModelOptimizeJobBuilder) Shell(shell string) *ModelOptimizeJobBuilder
Shell is used to specify linux shell type
func (*ModelOptimizeJobBuilder) TargetDevice ¶
func (m *ModelOptimizeJobBuilder) TargetDevice(targetDevice string) *ModelOptimizeJobBuilder
TargetDevice is used to specify model deploy device
func (*ModelOptimizeJobBuilder) Tolerations ¶
func (m *ModelOptimizeJobBuilder) Tolerations(tolerations []string) *ModelOptimizeJobBuilder
Tolerations is used to set tolerations for tolerate nodes,match option --toleration
type ModelProfileJobBuilder ¶
type ModelProfileJobBuilder struct { argsbuilder.ArgsBuilder // contains filtered or unexported fields }
func NewModelProfileJobBuilder ¶
func NewModelProfileJobBuilder() *ModelProfileJobBuilder
func (*ModelProfileJobBuilder) Annotations ¶
func (m *ModelProfileJobBuilder) Annotations(annotations map[string]string) *ModelProfileJobBuilder
Annotations is used to add annotations for job pods,match option --annotation
func (*ModelProfileJobBuilder) Build ¶
func (m *ModelProfileJobBuilder) Build() (*Job, error)
Build is used to build the job
func (*ModelProfileJobBuilder) CPU ¶
func (m *ModelProfileJobBuilder) CPU(cpu string) *ModelProfileJobBuilder
CPU assign cpu limits,match the option --cpu
func (*ModelProfileJobBuilder) Command ¶
func (m *ModelProfileJobBuilder) Command(args []string) *ModelProfileJobBuilder
Command is used to set job command
func (*ModelProfileJobBuilder) DataDirs ¶
func (m *ModelProfileJobBuilder) DataDirs(volumes map[string]string) *ModelProfileJobBuilder
DataDirs is used to mount host files to job containers,match option --data-dir
func (*ModelProfileJobBuilder) Datas ¶
func (m *ModelProfileJobBuilder) Datas(volumes map[string]string) *ModelProfileJobBuilder
Datas is used to mount k8s pvc to job pods,match option --data
func (*ModelProfileJobBuilder) Envs ¶
func (m *ModelProfileJobBuilder) Envs(envs map[string]string) *ModelProfileJobBuilder
Envs is used to set env of job containers,match option --env
func (*ModelProfileJobBuilder) GPUCore ¶
func (m *ModelProfileJobBuilder) GPUCore(core int) *ModelProfileJobBuilder
GPUCore is used to set gpu core for the job,match the option --gpucore
func (*ModelProfileJobBuilder) GPUCount ¶
func (m *ModelProfileJobBuilder) GPUCount(count int) *ModelProfileJobBuilder
GPUCount is used to set count of gpu for the job,match the option --gpus
func (*ModelProfileJobBuilder) GPUMemory ¶
func (m *ModelProfileJobBuilder) GPUMemory(memory int) *ModelProfileJobBuilder
GPUMemory is used to set gpu memory for the job,match the option --gpumemory
func (*ModelProfileJobBuilder) Image ¶
func (m *ModelProfileJobBuilder) Image(image string) *ModelProfileJobBuilder
Image is used to set job image,match the option --image
func (*ModelProfileJobBuilder) ImagePullPolicy ¶
func (m *ModelProfileJobBuilder) ImagePullPolicy(policy string) *ModelProfileJobBuilder
ImagePullPolicy is used to set image pull policy,match the option --image-pull-policy
func (*ModelProfileJobBuilder) ImagePullSecrets ¶
func (m *ModelProfileJobBuilder) ImagePullSecrets(secrets []string) *ModelProfileJobBuilder
ImagePullSecrets is used to set image pull secrests,match option --image-pull-secret
func (*ModelProfileJobBuilder) Inputs ¶
func (m *ModelProfileJobBuilder) Inputs(inputs string) *ModelProfileJobBuilder
Inputs is used to specify model inputs
func (*ModelProfileJobBuilder) Labels ¶
func (m *ModelProfileJobBuilder) Labels(labels map[string]string) *ModelProfileJobBuilder
Labels is used to add labels for job
func (*ModelProfileJobBuilder) Memory ¶
func (m *ModelProfileJobBuilder) Memory(memory string) *ModelProfileJobBuilder
Memory assign memory limits,match option --memory
func (*ModelProfileJobBuilder) ModelConfigFile ¶
func (m *ModelProfileJobBuilder) ModelConfigFile(filePath string) *ModelProfileJobBuilder
ModelConfigFile is used to set model config file,match the option --model-config-file
func (*ModelProfileJobBuilder) ModelName ¶
func (m *ModelProfileJobBuilder) ModelName(name string) *ModelProfileJobBuilder
ModelName is used to set model name,match the option --model-name
func (*ModelProfileJobBuilder) ModelPath ¶
func (m *ModelProfileJobBuilder) ModelPath(path string) *ModelProfileJobBuilder
ModelPath is used to set model path,match the option --model-path
func (*ModelProfileJobBuilder) Name ¶
func (m *ModelProfileJobBuilder) Name(name string) *ModelProfileJobBuilder
Name is used to set job name,match option --name
func (*ModelProfileJobBuilder) Namespace ¶
func (m *ModelProfileJobBuilder) Namespace(namespace string) *ModelProfileJobBuilder
Namespace is used to set job namespace,match option --namespace
func (*ModelProfileJobBuilder) NodeSelectors ¶
func (m *ModelProfileJobBuilder) NodeSelectors(selectors map[string]string) *ModelProfileJobBuilder
NodeSelectors is used to set node selectors for scheduling job,match option --selector
func (*ModelProfileJobBuilder) Outputs ¶
func (m *ModelProfileJobBuilder) Outputs(outputs string) *ModelProfileJobBuilder
Outputs is used to specify model outputs
func (*ModelProfileJobBuilder) ReportPath ¶
func (m *ModelProfileJobBuilder) ReportPath(reportPath string) *ModelProfileJobBuilder
ReportPath is used to specify profile result path
func (*ModelProfileJobBuilder) Shell ¶
func (m *ModelProfileJobBuilder) Shell(shell string) *ModelProfileJobBuilder
Shell is used to specify linux shell type
func (*ModelProfileJobBuilder) Tolerations ¶
func (m *ModelProfileJobBuilder) Tolerations(tolerations []string) *ModelProfileJobBuilder
Tolerations is used to set tolerations for tolerate nodes,match option --toleration