nflows icon indicating copy to clipboard operation
nflows copied to clipboard
verified publisher icon bayesiains

Added CUDA bug fix to the Sigmoid in the nonlinearities module

Open Donglin-Wang2 opened this issue 4 years ago • 0 comments

When we are setting the "learn_temperature" to False, the temperature is not sent to the correct device when trying to train on the GPU. By registering the temperature as a buffer, the temperature constant can be sent to the correct device.

Donglin-Wang2 avatar Aug 05 '21 13:08 Donglin-Wang2