jit

package

v1.11.0-0.1.5 Latest Latest Go to latest Published: Apr 2, 2023 License: MIT Imports: 5 Imported by: 2

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

This section is empty.

This section is empty.

This section is empty.

type JitModule struct {
	Pointer C.JitModule
}

A container for a torch script module in C++.

func Load(path string, device *torch.Device) (*JitModule, error)

Load the JIT module from the given path and map it onto a compute device.

func (module *JitModule) CastTo(dtype torch.Dtype) *JitModule

Cast the model's parameters to the given data-type in-place.

func (module *JitModule) CopyTo(device *torch.Device) *JitModule

Copy the model's parameters to the given compute accelerator in-place.

func (module *JitModule) Eval() *JitModule

Set the module to evaluation (e.g., inference) mode.

func (module *JitModule) Forward(inputs []*torch.IValue) (output *torch.IValue)

Forward pass IValues through the module and return the resulting IValue.

func (module *JitModule) IsTraining() bool

Return true if training features are enabled for the module, false otherwise.

func (module *JitModule) Save(path string) error

Save the module to the given path.

func (module *JitModule) String() string

Convert the module to a human-readable string representation.

func (module *JitModule) To(device *torch.Device, dtype torch.Dtype) *JitModule

Cast the model's parameters to the given data-type in-place.

func (module *JitModule) Train(mode bool) *JitModule

Enable/disable training features for the module.