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

fix(benchmark/camelyon): actually rely on gpu setup at Dependency level #244

Merged
merged 2 commits into from
Sep 3, 2024

Conversation

thbcmlowk
Copy link
Contributor

Related issue

# followed by the number of the issue

Summary

Notes

Please check if the PR fulfills these requirements

  • If the feature has an impact on the user experience, the changelog has been updated
  • Tests for the changes have been added (for bug fixes / features)
  • Docs have been added / updated (for bug fixes / features)
  • The commit message follows the conventional commit specification

@thbcmlowk thbcmlowk requested a review from a team as a code owner September 3, 2024 15:01
Copy link
Member

@ThibaultFy ThibaultFy left a comment

Choose a reason for hiding this comment

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

This argument is used on substra-ci and need to be updated at the same time

@@ -64,7 +64,7 @@ def fed_avg(params: dict, train_folder: Path, test_folder: Path):
mode=exp_params["mode"],
cp_name=exp_params["cp_name"],
cancel_cp=exp_params["cancel_cp"],
torch_gpu=exp_params["torch_gpu"],
Copy link
Member

@ThibaultFy ThibaultFy Sep 3, 2024

Choose a reason for hiding this comment

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

This will break the CI, need to do a companion PR on test-release-dev.yaml

Copy link
Contributor Author

@thbcmlowk thbcmlowk Sep 3, 2024

Choose a reason for hiding this comment

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

Yes, this is identified, and it will be updated - but thanks for the reminder!

Copy link
Contributor Author

Choose a reason for hiding this comment

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

@@ -108,6 +108,7 @@ def substrafl_fed_avg(
# Keeping editable_mode=True to ensure nightly test benchmarks are ran against main substrafl git ref
editable_mode=True,
compile=True,
use_gpu=use_gpu,
Copy link
Member

Choose a reason for hiding this comment

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

The renaming is not necessary, you can just set use_gpu=torch_gpu

Copy link
Contributor Author

@thbcmlowk thbcmlowk Sep 3, 2024

Choose a reason for hiding this comment

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

Yes, it was mainly for consistency through the workflow and to avoid having a third variable name for this. If you prefer keeping torch_gpu I can revert tho, I don't mind!

@thbcmlowk thbcmlowk force-pushed the fix/camelyon-with-gpu branch from 03b2fa2 to aff0f26 Compare September 3, 2024 15:52
Copy link
Member

@ThibaultFy ThibaultFy left a comment

Choose a reason for hiding this comment

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

If the companion PR is ready, I'm ok with the renaming :)

Thanks

Signed-off-by: Thibault Camalon <[email protected]>
@thbcmlowk thbcmlowk merged commit 92ddbf3 into main Sep 3, 2024
10 checks passed
@thbcmlowk thbcmlowk deleted the fix/camelyon-with-gpu branch September 3, 2024 16:17
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.

3 participants