vgg16

Documentation

Overview

Package vgg16 implements VGG16 network training.

Index

• type Benchmark
  • func NewBenchmark(driver *driver.Driver) *Benchmark
  • func (b *Benchmark) Run()
  • func (b *Benchmark) SelectGPU(gpuIDs []int)
  • func (b *Benchmark) SetUnifiedMemory()
  • func (b *Benchmark) Verify()

Types

type Benchmark

type Benchmark struct {
	BatchSize          int
	Epoch              int
	MaxBatchPerEpoch   int
	EnableTesting      bool
	EnableVerification bool
	// contains filtered or unexported fields
}

Benchmark defines the VGG16 network training benchmark.

func NewBenchmark

func NewBenchmark(driver *driver.Driver) *Benchmark

NewBenchmark creates a new benchmark.

func (*Benchmark) Run

func (b *Benchmark) Run()

Run executes the benchmark.

func (*Benchmark) SelectGPU

func (b *Benchmark) SelectGPU(gpuIDs []int)

SelectGPU selects the GPU to use.

func (*Benchmark) SetUnifiedMemory

func (b *Benchmark) SetUnifiedMemory()

SetUnifiedMemory asks the benchmark to use unified memory.

func (*Benchmark) Verify

func (b *Benchmark) Verify()

Verify runs the benchmark on the CPU and checks the result.