Clip by global norm type issue
lollcat opened this issue · comments
Using optax clip by global norm is giving type issue - the source of this should be found and fixed.
Fixed - was related to default type of hk.get_parameter
in real_nvp
.
lollcat opened this issue · comments
Using optax clip by global norm is giving type issue - the source of this should be found and fixed.
Fixed - was related to default type of hk.get_parameter
in real_nvp
.