As discussed here and previous messages, according to pytorch docs on model and optimizer. User should not move model to another device after constructing an optimizer for the model:
We need to handle this in the code:
cc @kai-tub
Hi, I would like to work on this and complete the work I've started. :)
When this issue is closed I will come back to my tiny_benchmark_wrapper.
@kai-tub sounds good, thanks !
One quick question, what do you mean by:
... add a warning note on changed behavior
Is a normal Note: environment in the docstrings sufficient? Or do you mean warning from import warnings?
Good question, I mean this
.. warning::
Be careful
which is rendered as here: https://pytorch.org/ignite/metrics.html#ignite.metrics.EpochMetric
Most helpful comment
Hi, I would like to work on this and complete the work I've started. :)
When this issue is closed I will come back to my
tiny_benchmark_wrapper.