neuralhelpers

package

v0.0.0-...-98db5b7 Latest Latest Go to latest Published: May 7, 2020 License: CC-BY-4.0 Imports: 14 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

Repository

github.com/dereklstinson/GoCuNets

Documentation ¶

Index ¶

func CreateRandCropBatch(input, output []image.Image, h, w, resizedouth, resizedoutw int) (tin, tout *cvdh.Tensor4d)
func Decoder(builder *gocunets.Builder, batchsize int32, outputchannel int32, ...) (mnet *gocunets.SimpleModuleNetwork)
func Encoder(builder *gocunets.Builder, batchsize int32, outputchannel int32, ...) (mnet *gocunets.SimpleModuleNetwork)
func GetImages(r []io.Reader) []image.Image
type GPULoader
- func CreateGpuLoader(dev cudart.Device, nbuffer int, SIBperbuff uint) (g *GPULoader, err error)

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

func CreateRandCropBatch ¶

func CreateRandCropBatch(input, output []image.Image, h, w, resizedouth, resizedoutw int) (tin, tout *cvdh.Tensor4d)

CreateRandCropBatch creates a RandCropBatch

func Decoder ¶

func Decoder(
	builder *gocunets.Builder,
	batchsize int32,
	outputchannel int32,
	hiddenoutputchannels []int32,
	learningrates, decay1, decay2 float32,
	x, dx *gocunets.Tensor) (mnet *gocunets.SimpleModuleNetwork)

Decoder decodes a vector

func Encoder ¶

func Encoder(
	builder *gocunets.Builder,
	batchsize int32,
	outputchannel int32,
	hiddenoutputchannels []int32,
	learningrates, decay1, decay2 float32,
	x *gocunets.Tensor) (mnet *gocunets.SimpleModuleNetwork)

Encoder encodes an image

func GetImages ¶

func GetImages(r []io.Reader) []image.Image

GetImages gets images

Types ¶

type GPULoader ¶

type GPULoader struct {
	// contains filtered or unexported fields
}

GPULoader loads the gpu

func CreateGpuLoader ¶

func CreateGpuLoader(dev cudart.Device, nbuffer int, SIBperbuff uint) (g *GPULoader, err error)

CreateGpuLoader creates a gpu loader

func (*GPULoader) AsyncLoadMem ¶

func (g *GPULoader) AsyncLoadMem(dest cutil.Pointer)

func (*GPULoader) LoadBuffer ¶

func (g *GPULoader) LoadBuffer(tensor *cvdh.Tensor4d)

func (*GPULoader) SyncStream ¶

func (g *GPULoader) SyncStream() error

Source Files ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL