Cornelius Emde
Cornelius Emde
My suggestion for the API would be to have arguments - `amp_backend` as one argument ("apex" vs "native") - `precision`, which is either [00, 01, 02, 03] for apex or...
> What I don't really like is that the product of options is very sparse. It does feel boilerplate-y to me but there's no better solution without going through deprecations....
@sonichi yes, as far as I remember, all that is necessary is the lightning callback to report training metrics for pruning.
@Borda I'll give it a go!
@awaelchli Good point. Maybe it would be good to immediately raise an exception when `save_hyperparameters` is not called from `__init__`?
I think it is quite a quick fix. Ill give it a go sometime this week.