probability Add get_config and from_config methods to DenseVariational

Add get_config and from_config methods to DenseVariational_v2

Open Frightera opened this issue 2 years ago • 2 comments

Fixes the issue #1627: NotImplementedError: Layer DenseVariational has arguments in __init__ and therefore must override get_config

Some unit tests are failing due to an internal issue in TF-Nightly as of 17/02/2023. Please find the gist here to see the fix.

Thanks!

Feb 17 '23 01:02 Frightera

Hi @csuter, is there any chance that you can take a look at this tiny PR?

Thanks.

May 15 '23 21:05 Frightera

thanks for sending.

please add some unit tests covering the new behavior, in the corresponding test file.

would any other layers benefit from adding the same new behavior?

May 19 '23 16:05 csuter