A-strategist
Results
1
comments of
A-strategist
I find the result layer from prune_low_magnitude would end up with different dtype_policy from the original one. And here is a solution: call _set_dtype_policy after prune_low_magnitude. for example ```python from...