Skip to content

Commit

Permalink
Update new-model-inference.yml
Browse files Browse the repository at this point in the history
  • Loading branch information
nanthan987 committed Apr 21, 2024
1 parent e0a1845 commit ba0f0a8
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/new-model-inference.yml
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ jobs:
echo ${{ secrets.USERPWD }} | sudo -S docker build -f Dockerfile -t "$DOCKER_IMAGE_NAME" .
echo ${{ secrets.USERPWD }} | sudo -S docker container run -m 8G --name "$DOCKER_IMAGE_NAME" --rm "$DOCKER_IMAGE_NAME":latest /bin/bash -c "ls"
echo ${{ secrets.USERPWD }} | sudo -S docker container run -m 8G --name "$DOCKER_IMAGE_NAME" --rm "$DOCKER_IMAGE_NAME":latest /bin/bash -c "ls work_dir"
./evaluate.sh .
echo ${{ secrets.USERPWD }} | ./evaluate.sh .
- name: Display accuracy, efficiency, and running time
run: |
Expand Down
2 changes: 1 addition & 1 deletion CVPR24_time_eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@
# To obtain the running time for each case, testing cases are inferred one-by-one
for case in test_cases:
shutil.copy(join(test_img_path, case), input_temp)
cmd = 'sudo docker container run -m 8G --name {} --rm -v {}:/workspace/inputs/ -v {}:/workspace/outputs/ {}:latest /bin/bash -c "sh predict.sh" '.format(teamname, input_temp, output_temp, teamname)
cmd = 'sudo -S docker container run -m 8G --name {} --rm -v {}:/workspace/inputs/ -v {}:/workspace/outputs/ {}:latest /bin/bash -c "sh predict.sh" '.format(teamname, input_temp, output_temp, teamname)
print(teamname, ' docker command:', cmd, '\n', 'testing image name:', case)
start_time = time.time()
os.system(cmd)
Expand Down

0 comments on commit ba0f0a8

Please sign in to comment.