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

Fixing slow pipeline tests (#14260)

Browse Source 
* Fiixng slow pipeline tests

* Remove the image-segmentaiton override.

* Fixing clamping only in training.

* Wav2vec2.

* Remove last mention of `no_grad`.

* Fixing copies.

* Rename.
This commit is contained in:
Nicolas Patry
2021-11-04 09:49:55 +01:00
committed by GitHub
parent 1a674ce679
commit 68427c9beb
7 changed files with 71 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_pipelines_audio_classification.py
View File

@@ -117,7 +117,7 @@ class AudioClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTest
self.assertEqual(
nested_simplify(output, decimals=4),
[
{"score": 0.9809, "label": "go"},
{"score": 0.981, "label": "go"},
{"score": 0.0073, "label": "up"},
{"score": 0.0064, "label": "_unknown_"},
{"score": 0.0015, "label": "down"},