Skip to main content

init_optim_kernel_cache

entrenar::autograd::cuda_optim

Function init_optim_kernel_cache 

Source 
pub fn init_optim_kernel_cache(ctx: Arc<CudaContext>) -> Result<()>
Expand description

Initialize optimizer kernel cache with CUDA context