diff --git a/Dockerfile b/Dockerfile index 9b13f7c5..7d078339 100644 --- a/Dockerfile +++ b/Dockerfile @@ -56,7 +56,7 @@ COPY . /usr/src/app # sudo nvidia-docker run --ipc=host --mount type=bind,source="$(pwd)"/coco,target=/usr/src/coco ultralytics/yolov3:v0 python3 train.py # Pull and Run with local directory access -# export tag=ultralytics/yolov3:v0 && sudo docker pull $tag && sudo nvidia-docker run --ipc=host --mount type=bind,source="$(pwd)"/coco,target=/usr/src/coco $tag python3 train.py +# export tag=ultralytics/yolov3:v0 && sudo docker pull $tag && sudo nvidia-docker run -it --ipc=host --mount type=bind,source="$(pwd)"/coco,target=/usr/src/coco $tag python3 train.py # Build and Push # export tag=ultralytics/yolov3:v0 && sudo docker build -t $tag . && sudo docker push $tag