what will be the closest replacement for @auto_move_data?
I really liked @auto_move_data as it gave me a nice and easy way to pass (a single) input to a model, without caring about moving data structures to devices, but now it will be removed 😢
I realize that you want people to use trainer.predict() but this requires a dataloader, right?, which at least for me oftentimes is overly complicated while developing.
As far as I see the dataloader/collate/transforms also have to be different from the normal ones, since these typically provide batches with inputs and targets, whereas the forward function only takes inputs.
Likely I'm missing something, can you maybe give some hints about this?
After writing this I realized that I can use something like this:
model = model.eval()
i, _ = next(iter(data.test_dataloader()))
with th.no_grad():
    outputs = model.forward(model.transfer_batch_to_device(i))

But I'm still curious about the intended future ways of doing this....