tfkeras

Documentation

Index

• Variables
• type Experiment
  • func Combination(models []Model, optimizers []KFOptimizer, batchSizes []int) []Experiment
  • func Default() []Experiment
  • func (e Experiment) Job(kfRoot string, strategy base.Strategy, hl plan.HostList, pr plan.PortRange, ...) job.Job
• type KFOptimizer
• type Model

Variables

var Models = []Model{
	MobileNetV2,
	ResNet50,
}

var Optimizers = []KFOptimizer{
	SyncSgd,
	SyncSgdNccl,
}

Types

type Experiment

type Experiment struct {
	Model     Model
	BatchSize int

	WarmupBatches   int
	NumIters        int
	NumBatchPerIter int

	KFOptimizer KFOptimizer
}

func Combination

func Combination(models []Model, optimizers []KFOptimizer, batchSizes []int) []Experiment

func Default

func Default() []Experiment

func (Experiment) Job

func (e Experiment) Job(kfRoot string, strategy base.Strategy, hl plan.HostList, pr plan.PortRange, logDir string) job.Job

type KFOptimizer

type KFOptimizer string

const (
	SyncSgd     KFOptimizer = `sync-sgd`
	SyncSgdNccl KFOptimizer = `sync-sgd-nccl`
)

type Model

type Model string

const (
	MobileNetV2 Model = `MobileNetV2`
	ResNet50    Model = `ResNet50`
)