plugins

Documentation

Index

• Variables
• func CreateDefaultTbaseEvalJobInput() (input *pb.JobInput)
• func CreateDefaultTbaseInferJobInput() (input *pb.JobInput)
• func CreateDefaultTbaseTrainJobInput() (input *pb.JobInput)
• func GetTbaseInstallRepoCmds(repo string, tag string) (cmds []string)
• type Plugin
  • func New(job *pb.Job) (p Plugin)
• type TbasePlugin
  • func NewTbasePlugin() (p *TbasePlugin)
  • func (p *TbasePlugin) GenerateCmds(input *pb.JobInput, id string) (cmds []string, err error)
  • func (p *TbasePlugin) ParseBar(encode string) (currentStep uint32, totalSteps uint32, err error)
  • func (p *TbasePlugin) ParseEval(encode string, jobId, modelId uint64) (out *pb.JobOutput, err error)
  • func (p *TbasePlugin) ParseInfer(encode, date string, jobId, modelId uint64) (out *pb.JobOutput, err error)

Variables

var EVAL_DIR = path.Join(ROOT_DATA_DIR, "evals")

var INFER_DIR = path.Join(ROOT_DATA_DIR, "infers")

var JOB_LOG_DIR = path.Join(ROOT_DATA_DIR, "logs")

var JOB_SHELL_DIR = path.Join(ROOT_DATA_DIR, "shells")

var MODEL_DIR = path.Join(ROOT_DATA_DIR, "models")

var PROGRESS_BAR_DIR = path.Join(ROOT_DATA_DIR, "progress_bars")

var ROOT_DATA_DIR = os.Getenv("RUNNER_DATA_DIR")

var TENSORBOARD_DIR = path.Join(ROOT_DATA_DIR, "tensorboards")

Functions

func CreateDefaultTbaseEvalJobInput

func CreateDefaultTbaseEvalJobInput() (input *pb.JobInput)

func CreateDefaultTbaseInferJobInput

func CreateDefaultTbaseInferJobInput() (input *pb.JobInput)

func CreateDefaultTbaseTrainJobInput

func CreateDefaultTbaseTrainJobInput() (input *pb.JobInput)

func GetTbaseInstallRepoCmds

func GetTbaseInstallRepoCmds(repo string, tag string) (cmds []string)

Types

type Plugin

type Plugin interface {
	GenerateCmds(input *pb.JobInput, id string) (cmds []string, err error)
	ParseBar(encode string) (currentStep uint32, totalSteps uint32, err error)
	ParseEval(encode string, jobId, modelId uint64) (out *pb.JobOutput, err error)
	ParseInfer(encode, date string, jobId, modelId uint64) (out *pb.JobOutput, err error)
}

func New

func New(job *pb.Job) (p Plugin)

type TbasePlugin

type TbasePlugin struct {
	Sep string
}

func NewTbasePlugin

func NewTbasePlugin() (p *TbasePlugin)

func (*TbasePlugin) GenerateCmds

func (p *TbasePlugin) GenerateCmds(input *pb.JobInput, id string) (cmds []string, err error)

func (*TbasePlugin) ParseBar

func (p *TbasePlugin) ParseBar(encode string) (currentStep uint32, totalSteps uint32, err error)

func (*TbasePlugin) ParseEval

func (p *TbasePlugin) ParseEval(encode string, jobId, modelId uint64) (out *pb.JobOutput, err error)

func (*TbasePlugin) ParseInfer

func (p *TbasePlugin) ParseInfer(encode, date string, jobId, modelId uint64) (out *pb.JobOutput, err error)