This commit is contained in:
Sam Shleifer 2020-07-15 14:59:38 -04:00 коммит произвёл GitHub
Родитель b23d3a5ad4
Коммит 1a647abf0b
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
2 изменённых файлов: 0 добавлений и 7 удалений

Просмотреть файл

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

Просмотреть файл

@ -143,12 +143,6 @@ class T5ModelTester:
decoder_attention_mask=decoder_attention_mask,
)
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(decoder_output.size(), (self.batch_size, self.decoder_seq_length, self.hidden_size))
self.parent.assertEqual(len(decoder_past), 2)