SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update self-push.yml (#13364)

Browse Source
This commit is contained in:
Patrick von Platen
2021-09-01 09:37:51 +02:00
committed by GitHub
parent c02cd95c56
commit 8e20887886
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/self-push.yml vendored
View File

@@ -79,7 +79,7 @@ jobs:
path: reports path: reports
run_tests_flax_gpu: run_tests_flax_gpu:
runs-on: [self-hosted, docker-gpu, single-gpu] runs-on: [self-hosted, docker-gpu-test, single-gpu]
container: container:
image: tensorflow/tensorflow:2.4.1-gpu image: tensorflow/tensorflow:2.4.1-gpu
options: --gpus 0 --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ options: --gpus 0 --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/