types

Documentation

Index

• Constants
• type FrameworkType
• type Job
• type Parameter

Constants

const (
	// FrameworkTypeTensorFlow defines tensorflow type.
	FrameworkTypeTensorFlow = "tensorflow"
	// FrameworkTypePyTorch defines pytorch type.
	FrameworkTypePyTorch = "pytorch"
)

const (
	CleanPodPolicyAll     = "all"
	CleanPodPolicyRunning = "running"
	CleanPodPolicyNone    = "none"
)

CleanPodPolicy describes how to deal with pods when the job is finished.

Types

type FrameworkType

type FrameworkType string

FrameworkType is the type for types of the frameworks.

type Job

type Job struct {
	Framework FrameworkType
	Name      string
	PS        int
	// Master is for PyTorchJob.
	Master int
	Worker int
}

Job is the type for general jobs (PyTorchJob, TFJob).

type Parameter

type Parameter struct {
	Framework   FrameworkType
	PSCount     int
	WorkerCount int
	// MasterCount is for PyTorchJob.
	MasterCount  int
	GenerateName string
	Image        string
	CleanPolicy  string
	Resource     resource.JobResource
}

Parameter is the type for the parameter of the job.