Skip to content

Commit 03d5ee3

Browse files
Reduce memory requirement for pytorch job
1 parent a3ff979 commit 03d5ee3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/kfto/core/kfto_training_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,12 +164,12 @@ func createKFTOPyTorchJob(test Test, namespace string, config corev1.ConfigMap,
164164
Resources: corev1.ResourceRequirements{
165165
Requests: corev1.ResourceList{
166166
corev1.ResourceCPU: resource.MustParse("2"),
167-
corev1.ResourceMemory: resource.MustParse("16Gi"),
167+
corev1.ResourceMemory: resource.MustParse("8Gi"),
168168
corev1.ResourceName(gpuLabel): resource.MustParse(fmt.Sprint(numGpus)),
169169
},
170170
Limits: corev1.ResourceList{
171171
corev1.ResourceCPU: resource.MustParse("2"),
172-
corev1.ResourceMemory: resource.MustParse("16Gi"),
172+
corev1.ResourceMemory: resource.MustParse("8Gi"),
173173
corev1.ResourceName(gpuLabel): resource.MustParse(fmt.Sprint(numGpus)),
174174
},
175175
},

0 commit comments

Comments
 (0)