You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
When ray train is performed, the train_loop_per_worker function has a return value. How can I get the return value of the train_loop_per_worker function after running the trainer.fit()
The text was updated successfully, but these errors were encountered:
Hxinyue
added
bug
Something that is supposed to be working; but isn't
triage
Needs triage (eg: priority, bug/not-bug, and owning component)
labels
Jan 8, 2025
What happened + What you expected to happen
When ray train is performed, the train_loop_per_worker function has a return value. How can I get the return value of the train_loop_per_worker function after running the trainer.fit()
Versions / Dependencies
ray==2.11.0
python==3.10
Reproduction script
def main(config):
alg_id = config["alg_id"]
return {"success": 1}
trainer = TorchTrainer(
main,
scaling_config=ScalingConfig(use_gpu=use_gpu, num_workers=num_workers,
resources_per_worker={"CPU": cpu, "GPU": gpu}),
train_loop_config=config,
...
)
trainer.fit()
Issue Severity
High: It blocks me from completing my task.
The text was updated successfully, but these errors were encountered: