optimizerBuilder property
A callback that builds the optimizer once all layer parameters are allocated in VRAM.
Implementation
OptimizerGPU Function(List<GPUTensor>)? optimizerBuilder;
A callback that builds the optimizer once all layer parameters are allocated in VRAM.
OptimizerGPU Function(List<GPUTensor>)? optimizerBuilder;