Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Remove TrainParams #55

Merged
merged 1 commit into from
Feb 12, 2023
Merged

Remove TrainParams #55

merged 1 commit into from
Feb 12, 2023

Conversation

norabelrose
Copy link
Member

Simplifies the training code a little bit, removing an internal dataclass

@@ -141,31 +118,29 @@ def fit(
if verbose:
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should we maybe remove verbose also from the css class, and just always print the things under the ifs ?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hmm I think we should either keep verbose or never print, because I want to support programmatic uses where you may not want to print every time you call fit. Like imagine if PyTorch optimizers always printed when you initialized them or called step or something.

@norabelrose norabelrose merged commit d388de7 into main Feb 12, 2023
@norabelrose norabelrose deleted the train-simplify branch February 12, 2023 18:32
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.

2 participants