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

TF GPU CI (#3085)

Browse Source 
* debug env

* Restrict TF GPU memory

* Fixup

* One more test

* rm debug logs

* Fixup
This commit is contained in:
Julien Chaumond
2020-03-02 15:45:25 -05:00
committed by GitHub
parent d3eb7d23a4
commit f169957d0c
5 changed files with 35 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -40,7 +40,8 @@ jobs:
- name: Run all non-slow tests on GPU
env:
TF_FORCE_GPU_ALLOW_GROWTH: yes
TF_FORCE_GPU_ALLOW_GROWTH: "true"
# TF_GPU_MEMORY_LIMIT: 4096
OMP_NUM_THREADS: 1
USE_CUDA: yes
run: |

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

@@ -41,7 +41,7 @@ jobs:
- name: Run all tests on GPU
env:
TF_FORCE_GPU_ALLOW_GROWTH: yes
TF_FORCE_GPU_ALLOW_GROWTH: "true"
OMP_NUM_THREADS: 1
RUN_SLOW: yes
USE_CUDA: yes