Function std::cuda_half_sgd

#include <std/ml/gpu_cuda.h> source · definition

int cuda_half_sgd(struct CudaHalfBuf w, struct CudaHalfBuf g, float lr)

w = round(w - lr * g)