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

done (#16012)

Browse Source
This commit is contained in:
Francesco Saverio Zuppichini
2022-03-09 15:51:56 +01:00
committed by GitHub
parent 38bce1d4cf
commit 1e8f37992f
2 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/maskformer/test_modeling_maskformer.py
View File

@@ -139,7 +139,7 @@ class MaskFormerModelTester:
def comm_check_on_output(result):
# let's still check that all the required stuff is there
self.parent.assertTrue(result.transformer_decoder_hidden_states is not None)
self.parent.assertTrue(result.transformer_decoder_last_hidden_state is not None)
self.parent.assertTrue(result.pixel_decoder_last_hidden_state is not None)
self.parent.assertTrue(result.encoder_last_hidden_state is not None)
# okay, now we need to check the logits shape