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
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!
@amogkam Is there an end to end example somewhere?
yes you can find them here Ray Train Examples — Ray v1.10.0