Deep Learning GPU CODER
2 次查看(过去 30 天)
显示 更早的评论
I have question about Deep Learning Training, GPU Coder and Precision
- We are working on a project that trains data on pre-trained models such as Resnet50 and AlexNet. I would like to use GPU CODER because I heard that using Tensor Core can achieve faster speed efficiency. However, when I looked at the examples related to GPU CODER, there were only examples of data validation using the model. I wonder if it is impossible to train data on a pre-trained model using GPU CODER. If possible, I'd like to ask if there are any relevant examples
- I know that MATLAB uses Single Precision by default when Deep Learning Training. How to use Double Precision or Half Precision?
0 个评论
采纳的回答
Sayan Saha
2022-5-12
Hi Kyle,
GPU Coder only supports inference using deep learning networks currently. Training the network using coder products is not yet supported. Given that you are using pre-trained networks, you can leverage transfer learning capability in MATLAB to re-train the networks for your datasets.
Training the network can only be done using single precision in MATLAB currently. Can you elaborate on your requirements to train using double and half precision? I understand that half-precision is likely to improve the training time. Are there any other reasons?
For inference we do support quantized datatype with CuDNN and TensorRT libraries as documented in: https://www.mathworks.com/help/deeplearning/ug/quantize-a-residual-network-trained-for-image-classification-and-generate-cuda-code.html
https://www.mathworks.com/help/gpucoder/ug/tensorrt-target.html
With TensorRT library half-precision is also supported: https://www.mathworks.com/help/gpucoder/ug/code-generation-using-tensorrt.html#mw_c1f34b7f-7d7a-4fd8-ad98-be0d4212118e
~Sayan
更多回答(0 个)
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Kernel Creation from MATLAB Code 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!