accelerate

take `torch.nn.Module` model into account when moving to device

#3167

Merged