Ray train usage?

Hi – I am trying to follow the example here: Ray Train User Guide — Ray v1.10.0 and find that train.torch is missing. Trying to find the equivalent of train.torch.prepare_model, etc. Any guidance here?

Regards,
RAghu

ah yes the example is missing this line

import ray.train.torch. If you add that, the example should work!

1 Like

@amogkam Is there an end to end example somewhere?

yes you can find them here Ray Train Examples — Ray v1.10.0

1 Like