Parcourir la source

Update pytorch-deploy.yaml

John Lockman il y a 3 ans
Parent
commit
b88359c792
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      examples/PyTorch/pytorch-deploy.yaml

+ 1 - 1
examples/PyTorch/pytorch-deploy.yaml

@@ -12,7 +12,7 @@ spec:
         volumeMounts:
         - mountPath: /pyscript
           name: torch-job-volume
-        command: ["bash","-c","python /pyscript/pytorchcpu-example.py"]
+        command: ["bash","-c","python /pyscript/pytorch-example.py"]
       restartPolicy: Never
       volumes:
       - name: torch-job-volume