model

Jun 7, 2021

Changes in this version

+ const TargetDockerCPU

+ const TargetDockerGPU

+ var CUDABaseImages []CUDABaseImage

+ var TFCompatibilityMatrix []TFCompatibility

+ var TorchCompatibilityMatrix []TorchCompatibility

+ func CUDABaseImageFor(cuda string, cuDNN string) (string, error)

+ type ArgumentType string

+ const ArgumentTypeBool

+ const ArgumentTypeFloat

+ const ArgumentTypeInt

+ const ArgumentTypePath

+ const ArgumentTypeString

+ type CUDABaseImage struct

+ CUDA string

+ CuDNN string

+ IsDevel bool

+ Tag string

+ Ubuntu string

+ func (i *CUDABaseImage) ImageTag() string

+ func (i *CUDABaseImage) UnmarshalJSON(data []byte) error

+ type Config struct

+ Environment *Environment

+ Examples []*Example

+ Model string

+ Workdir string

+ func ConfigFromYAML(contents []byte) (*Config, error)

+ func DefaultConfig() *Config

+ func (c *Config) CUDABaseImageTag() (string, error)

+ func (c *Config) HasCPU() bool

+ func (c *Config) HasGPU() bool

+ func (c *Config) PythonPackagesForArch(arch string, goos string, goarch string) (packages []string, indexURLs []string, err error)

+ func (c *Config) ValidateAndCompleteConfig() error

+ type Environment struct

+ Architectures []string

+ BuildRequiresGPU bool

+ CUDA string

+ CuDNN string

+ PreInstall []string

+ PythonExtraIndexURLs []string

+ PythonFindLinks []string

+ PythonPackages []string

+ PythonRequirements string

+ PythonVersion string

+ SystemPackages []string

+ type Example struct

+ Input map[string]string

+ Output string

+ type Image struct

+ Arch string

+ BuildFailed bool

+ Created time.Time

+ RunArguments map[string]*RunArgument

+ TestStats *Stats

+ URI string

+ func ImageForArch(images []*Image, arch string) *Image

+ type Model struct

+ Host string

+ Name string

+ User string

+ func (m *Model) String() string

+ type RunArgument struct

+ Default *string

+ Help *string

+ Max *string

+ Min *string

+ Options *[]string

+ Type ArgumentType

+ type Stats struct

+ BootTime float64

+ CPUUsage float64

+ MemoryUsage uint64

+ RunTime float64

+ SetupTime float64

+ type TFCompatibility struct

+ CUDA string

+ CuDNN string

+ Pythons []string

+ TF string

+ TFCPUPackage string

+ TFGPUPackage string

+ func (compat *TFCompatibility) UnmarshalJSON(data []byte) error

+ type TorchCompatibility struct

+ CUDA *string

+ IndexURL string

+ Pythons []string

+ Torch string

+ Torchaudio string

+ Torchvision string

+ func (c *TorchCompatibility) TorchVersion() string

+ func (c *TorchCompatibility) TorchvisionVersion() string

+ type Version struct

+ BuildIDs map[string]string

+ Config *Config

+ Created time.Time

+ ID string