Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -35,4 +35,4 @@ COPY . /workspace
|
|
| 35 |
WORKDIR /workspace
|
| 36 |
|
| 37 |
# Set the command to run your training script
|
| 38 |
-
CMD ["python", "
|
|
|
|
| 35 |
WORKDIR /workspace
|
| 36 |
|
| 37 |
# Set the command to run your training script
|
| 38 |
+
CMD ["python", "train.py"]
|