Skip to content

Deprecate .dtype from all layers and modules, switch to .param_dtype

Vicentini Filippo requested to merge pv/flax-dev into master

Created by: PhilipVinc

This is to be consistent with the new flax. It's annoying, but it's for the better.

It will be mildly annoying for users by trying to be loud, but we don't break code because everything is properly deprecated.

I kind-of-like it as I prefer RBMModPhase(param_dtype=float32)to RBMModPhase(dtype=float32) to use single-precision.

Based on top of #1198

Merge request reports