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):
model = CTRLLMHeadModel.from_pretrained("ctrl")
input_ids = torch.tensor(
[[11858, 586, 20984, 8]], dtype=torch.long, device=torch_device
) # Legal My neighbor is
[[11859, 0, 1611, 8]], dtype=torch.long, device=torch_device
) # Legal the president is
expected_output_ids = [
11859,
586,
20984,
0,
1611,
8,
13391,
3,
980,
8258,
72,
327,
148,
5,
150,
26449,
2,
53,
29,
226,
19,
348,
469,
3,
780,
49,
3,
980,
] # Legal My neighbor is refusing to pay rent after 2 years and we are having to force him to pay
2595,
48,
20740,
246533,
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)
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
def test_lm_generate_ctrl(self):
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 = [
11859,
586,
20984,
0,
1611,
8,
13391,
3,
980,
8258,
72,
327,
148,
5,
150,
26449,
2,
53,
29,
226,
19,
348,
469,
3,
780,
49,
3,
980,
] # Legal My neighbor is refusing to pay rent after 2 years and we are having to force him to pay
2595,
48,
20740,
246533,
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)
self.assertListEqual(output_ids[0].tolist(), expected_output_ids)
self.assertListEqual(output_ids[0].numpy().tolist(), expected_output_ids)