ray.train.torch.prepare\_model ============================== .. currentmodule:: ray.train.torch .. autofunction:: prepare_model