CNN Performance: CPU Consistency vs. GPU Variance - Why?

1 visualización (últimos 30 días)
Hamza
Hamza el 22 de Nov. de 2023
Respondida: Ruth el 23 de Nov. de 2023
Hello everyone,
I have executed CNN code multiple times using rng(0) with CPU and consistently obtained the same result. However, when I attempted to accelerate the training process using the GPU, the results differed. Has anyone else faced this issue?
Thank you in advance!

Respuesta aceptada

Ruth
Ruth el 23 de Nov. de 2023
Hi Hamza,
Even when using "gpurng" some small non-deterministic behavior is expected to happen in the GPU during training, particularly during the backward pass. This is out of our control.
However the behavior should be deterministic in the forward pass and subsequently at prediction time.
If one sets the learning rate to be almost zero (e.g. 1e-16, meaning nothing is updated in the backward pass), the output of training (using "rng" and "gpurng") should look deterministic.
Best wishes,
Ruth

Más respuestas (1)

Edric Ellis
Edric Ellis el 23 de Nov. de 2023
I'm not certain if it will make everything consistent, but note that random state on the GPU is controlled by the gpurng function.

Categorías

Más información sobre Image Data Workflows en Help Center y File Exchange.

Community Treasure Hunt

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

Start Hunting!

Translated by