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

Upgrade to pytorch-lightning >=2 #114

Merged
merged 6 commits into from
Sep 19, 2024
Merged

Conversation

jamt9000
Copy link
Collaborator

@jamt9000 jamt9000 commented Aug 17, 2024

Upgrades pytorch lightning, modifying usage of pl.Trainer to conform with the new way to set the devices and checkpoint.

Also updates the readme commands and script for preparing the data for training.

Upgrades pytorch lightning, modifying usage of pl.Trainer
to conform with the new way to set the devices and checkpoint.
Include the commands for unzipping the files, including the zipped
csvs within the original zip from the first challenge.
See if we can use the mac GPU for the trainer test
Avoids trying to use MPS on mac in CI which has insufficient memory
In preprocessing_utils.py save the created val.csv in the
same folder as the input csv instead of in the current working
directory, and add logging so the user knows where it is saved.

This makes the two functions in the file more consistent in where
they save their output.
@jamt9000 jamt9000 requested a review from m13uz August 20, 2024 13:15
@jamt9000 jamt9000 merged commit 0eabf50 into master Sep 19, 2024
5 checks passed
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.

1 participant