Speed of custom training loop

4 次查看（过去 30 天）
Sangmin Lee
Sangmin Lee 2020-9-2
回答： Raynier Suresh 2021-2-25
Hi,
I test several code using 'Train Network Using Custom Training Loop'. I found it is severely slower than the method using 'nnet.cnn.layer.Layer ' class layer and 'trainNetwork' function.
Is it ture that 'dlnetwork' class is slower than 'nnet.cnn.layer.Layer ' class? or is it because of the example is using for-loop explicitly instead of using the built-in 'trainNetwork' function?
If I am wrong, please correct me or it would be great to improve speed of algorithm.

请先登录，再回答此问题。

回答（1 个）

Raynier Suresh
Raynier Suresh 2021-2-25
Hi, Generally trainNetwork framework is faster, it is optimised to take care of many things. But it's not that the custom training loop will always be slower. You can use the minibatchqueue, gpuArray and dlArray to speed things. Typically, if it is possible to use trainNetwork prefer using this.
If you can provide more information about your workflow and the hardware you were using it will be helpful for us to investigate more on this.

类别

在 Help Center 和 File Exchange 中查找有关 Custom Training Loops 的更多信息

标签

产品


版本

R2019b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by