Issue with traced models

If user passes a traced model into create_supervised* methods and specifies a device, there will be the following error:

traced_model = torch.jit.trace(model, z)

evaluator = create_supervised_evaluator(traced_model, metrics, device='cuda')
> RuntimeError: to is not supported on TracedModules

We need to check if model have the method .to…

[vfdev-5]

@TheCodez you can do it like this:

# Check if torch-nightly
if "dev" in torch.__version__:
     ...
else:
    ...

[vfdev-5]

Probably, we should wait until this issue is done.

EDIT: https://github.com/pytorch/pytorch/pull/15340