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

[fix] check code quality (#5772)

Browse Source
This commit is contained in:
Sam Shleifer
2020-07-15 14:59:38 -04:00
committed by GitHub
parent b23d3a5ad4
commit 1a647abf0b
2 changed files with 0 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/seq2seq/utils.py
View File

@@ -182,7 +182,6 @@ class SortishSampler(Sampler):
return iter(sort_idx) return iter(sort_idx)
logger = getLogger(__name__) logger = getLogger(__name__)

6
tests/test_modeling_t5.py
View File

@@ -143,12 +143,6 @@ class T5ModelTester:
decoder_attention_mask=decoder_attention_mask, decoder_attention_mask=decoder_attention_mask,
) )
decoder_output, decoder_past, encoder_output = model(input_ids=input_ids, decoder_input_ids=decoder_input_ids) decoder_output, decoder_past, encoder_output = model(input_ids=input_ids, decoder_input_ids=decoder_input_ids)
result = {
"encoder_output": encoder_output,
"decoder_output": decoder_output,
"decoder_past": decoder_past,
}
self.parent.assertEqual(encoder_output.size(), (self.batch_size, self.encoder_seq_length, self.hidden_size)) self.parent.assertEqual(encoder_output.size(), (self.batch_size, self.encoder_seq_length, self.hidden_size))
self.parent.assertEqual(decoder_output.size(), (self.batch_size, self.decoder_seq_length, self.hidden_size)) self.parent.assertEqual(decoder_output.size(), (self.batch_size, self.decoder_seq_length, self.hidden_size))
self.parent.assertEqual(len(decoder_past), 2) self.parent.assertEqual(len(decoder_past), 2)