config

Documentation

Index

• type AICCConfig
• type Config
  • func LoadConfig(path string) (*Config, error)
  • func (cfg *Config) SetDefault()
  • func (cfg *Config) Validate() error
• type FinetuneConfig
• type ModelConfig
• type OBSConfig
• type UploadConfig
  • func (c *UploadConfig) SetDefault()

Types

type AICCConfig

type AICCConfig struct {
	Domain       string `json:"domain" required:"true"`
	User         string `json:"user" required:"true"`
	Password     string `json:"password" required:"true"`
	Project      string `json:"project" required:"true"`
	ProjectId    string `json:"project_id" required:"true"`
	AuthEndpoint string `json:"auth_endpoint" required:"true"`

	// modelarts endpoint
	Endpoint string `json:"endpoint" required:"true"`
}

type Config

type Config struct {
	Watch    watchimpl.Config `json:"watch"        required:"true"`
	Finetune FinetuneConfig   `json:"finetune"     required:"true"`
	AICC     AICCConfig       `json:"aicc"         required:"true"`
	Upload   UploadConfig     `json:"upload"         required:"true"`
	OBS      OBSConfig        `json:"obs"         required:"true"`
}

func LoadConfig

func LoadConfig(path string) (*Config, error)

func (*Config) SetDefault

func (cfg *Config) SetDefault()

func (*Config) Validate

func (cfg *Config) Validate() error

type FinetuneConfig

type FinetuneConfig struct {
	WukongConfig ModelConfig `json:"wukong"`
}

type ModelConfig

type ModelConfig struct {
	PoolId     string `json:"pool_id"`
	PoolName   string `json:"pool_name"`
	FlavorId   string `json:"flavor_id"`
	OutputKey  string `json:"output_key"`
	OutputDir  string `json:"output_dir"`
	WorkingDir string `json:"working_dir"`
	CodeDir    string `json:"code_dir"`
	InputDir   string `json:"input_dir"`
	LogDir     string `json:"log_dir"`
	ModelDir   string `json:"ckpt_file"`
	ImageURL   string `json:"image_url"`
}

type OBSConfig

type OBSConfig struct {
	AccessKey string `json:"access_key"    required:"true"`
	SecretKey string `json:"secret_key"    required:"true"`
	Endpoint  string `json:"endpoint"      required:"true"`
	Bucket    string `json:"bucket"        required:"true"`
}

type UploadConfig

type UploadConfig struct {
	UploadWorkDir     string `json:"upload_work_dir"      required:"true"`
	UploadFolderShell string `json:"upload_folder_shell"  required:"true"`

	// DownloadExpiry specifies the timeout to download a obs file.
	// The unit is second.
	DownloadExpiry int    `json:"download_expiry"`
	OBSUtilPath    string `json:"obsutil_path"             required:"true"`
}

func (*UploadConfig) SetDefault

func (c *UploadConfig) SetDefault()