godweiyang/NN-CUDA-Example · AI Repo StatsGOgodweiyang/NN-CUDA-Example
Several simple examples for popular neural network toolkits calling custom CUDA operators.
Huấn luyện & fine-tunePython Apache-2.0 206tạo 2021-03-18 #cpp#cuda#neural-network#python#pytorch#tensorflow