Skip to content

Conversation

@Yuyutsu01
Copy link

This PR improves the documentation for the place_model_on_device parameter
in the Trainer class.

The previous description was ambiguous and did not reflect the actual behavior.
This update clarifies that:

  • When True (default), the Trainer automatically places the model on the target device.
  • When False, the user must manually move the model (e.g., model.to(device)).

Fixes #29469.

…e#29469)Clarified the behavior of the 'place_model_on_device' parameter in Trainer.The previous description was ambiguous and did not reflect actual device handling.

Signed-off-by: yuyutsu01 <[email protected]>
@Rocketknight1
Copy link
Member

I'm not sure why this PR exists! That issue was closed over a year ago. Closing for now.

@Yuyutsu01
Copy link
Author

Yuyutsu01 commented Dec 2, 2025 via email

@Rocketknight1
Copy link
Member

Haha, don't be discouraged from contributing more, in that case! Just check when an issue was created - most older issues, even if they're still open, probably aren't relevant anymore.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Documentation Inconsistency for Trainer.place_model_on_device Parameter

3 participants