-
Notifications
You must be signed in to change notification settings - Fork 4
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
Conversation
There was a problem hiding this 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"], |
There was a problem hiding this comment.
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
There was a problem hiding this comment.
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!
There was a problem hiding this comment.
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, |
There was a problem hiding this comment.
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
There was a problem hiding this comment.
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!
… level Signed-off-by: Thibault Camalon <[email protected]>
03b2fa2
to
aff0f26
Compare
There was a problem hiding this 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]>
Related issue
#
followed by the number of the issueSummary
Notes
Please check if the PR fulfills these requirements