adamw_kernel.cu 11.2 KB