accelerate

87732a4c - take `torch.nn.Module` model into account when moving to device (#3167)