What is the default mini-batch size in the fitrnet function

17 次查看(过去 30 天)
Hi,
I have recently been using the fitrnet function from the Statistics and Machine learning toolbox in order to do regression. Although having used it for a while, I am not sure of the mini-batch size that is used when computing the gradient for training.
I have looked through the documentation for the fitrnet function but have not been able to find the answer. It does speak of iterations, but not the number of iterations per epoch.
Does anyone know where to get this information and if the function uses full-batchs, mini-batches or stochastic GD.
Thanks

采纳的回答

Rahul
Rahul 2022-10-12
编辑:Rahul 2022-10-12
Looking into "fitrnet.m" and verbose data there seems to be no explicit use of “mini batch size” and function is applied on entire “xdata” input variable, so the default mini batch size can be taken as 1.
But in custom training loops with "minibatchqueue" or "training options" the default “mini batch size” is 128.You can refer the following link for more information:
This is mostly used in custom training loops or while training deep learning architectures.

更多回答(0 个)

类别

Help CenterFile Exchange 中查找有关 Image Data Workflows 的更多信息

Community Treasure Hunt

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

Start Hunting!

Translated by