ray.train.trainer.BaseTrainer.setup — Ray 2.45.0 (original) (raw)

• ray.train.tr...

BaseTrainer.setup() → None[source]#

Called during fit() to perform initial setup on the Trainer.

Note

This method is run on a remote process.

This method will not be called on the driver, so any expensive setup operations should be placed here and not in __init__.

This method is called prior to preprocess_datasets andtraining_loop.