-
Notifications
You must be signed in to change notification settings - Fork 44
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
Update kfto test support disconnected #283
base: main
Are you sure you want to change the base?
Update kfto test support disconnected #283
Conversation
[APPROVALNOTIFIER] This PR is NOT APPROVED This pull-request has been approved by: The full list of commands accepted by this bot can be found here.
Needs approval from an approver in each of these files:
Approvers can indicate their approval by writing |
@@ -31,7 +31,7 @@ import ( | |||
) | |||
|
|||
func TestPyTorchJobWithCuda(t *testing.T) { | |||
runKFTOPyTorchJob(t, GetCudaTrainingImage(), "nvidia.com/gpu", 1) | |||
runKFTOPyTorchJob(t, GetCudaTrainingImage(), "nvidia.com/gpu", 0) |
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.
Why you have set numGpus to 0?
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.
Ahh, my mistake..
Thank you Shilpa for pointing that out !!
c45a274
to
1553b78
Compare
Description
RHOAIENG-16198
It basically downloads the alpaca-dataset from s3/minio and load it using load_from_disk method :
How Has This Been Tested?
Same python script tested locally :
Merge criteria: