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

test ctrl

Browse Source
This commit is contained in:
Patrick von Platen
2020-03-09 13:58:01 +00:00
parent b73dd1a0e4
commit b12541c4dc
2 changed files with 39 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
tests/test_modeling_ctrl.py
View File

@@ -220,30 +220,30 @@ class CTRLModelLanguageGenerationTest(unittest.TestCase):
def test_lm_generate_ctrl(self): def test_lm_generate_ctrl(self):
model = CTRLLMHeadModel.from_pretrained("ctrl") model = CTRLLMHeadModel.from_pretrained("ctrl")
input_ids = torch.tensor( input_ids = torch.tensor(
[[11858, 586, 20984, 8]], dtype=torch.long, device=torch_device [[11859, 0, 1611, 8]], dtype=torch.long, device=torch_device
) # Legal My neighbor is ) # Legal the president is
expected_output_ids = [ expected_output_ids = [
11859, 11859,
586, 0,
20984, 1611,
8, 8,
13391, 5,
3, 150,
980, 26449,
8258,
72,
327,
148,
2, 2,
53, 19,
29, 348,
226, 469,
3, 3,
780, 2595,
49, 48,
3, 20740,
980, 246533,
] # Legal My neighbor is refusing to pay rent after 2 years and we are having to force him to pay 246533,
19,
30,
5,
] # Legal the president is a good guy and I don't want to lose my job. \n \n I have a
output_ids = model.generate(input_ids, do_sample=False) output_ids = model.generate(input_ids, do_sample=False)
self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids) self.assertListEqual(output_ids[0].tolist(), expected_output_ids)

38
tests/test_modeling_tf_ctrl.py
View File

@@ -209,29 +209,29 @@ class TFCTRLModelLanguageGenerationTest(unittest.TestCase):
@slow @slow
def test_lm_generate_ctrl(self): def test_lm_generate_ctrl(self):
model = TFCTRLLMHeadModel.from_pretrained("ctrl") model = TFCTRLLMHeadModel.from_pretrained("ctrl")
input_ids = tf.convert_to_tensor([[11858, 586, 20984, 8]], dtype=tf.int32) input_ids = tf.convert_to_tensor([[11859, 0, 1611, 8]], dtype=tf.int32) # Legal the president is
expected_output_ids = [ expected_output_ids = [
11859, 11859,
586, 0,
20984, 1611,
8, 8,
13391, 5,
3, 150,
980, 26449,
8258,
72,
327,
148,
2, 2,
53, 19,
29, 348,
226, 469,
3, 3,
780, 2595,
49, 48,
3, 20740,
980, 246533,
] # Legal My neighbor is refusing to pay rent after 2 years and we are having to force him to pay 246533,
19,
30,
5,
] # Legal the president is a good guy and I don't want to lose my job. \n \n I have a
output_ids = model.generate(input_ids, do_sample=False) output_ids = model.generate(input_ids, do_sample=False)
self.assertListEqual(output_ids[0].tolist(), expected_output_ids) self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids)