()
| 20 | return old_impl(input_ids, **kw_args) + self.net(input_ids, input_ids,input_ids) |
| 21 | |
| 22 | def test_create(): |
| 23 | a = NestedModel(args=NestedModel.get_args(num_layers=2, |
| 24 | bert_args={"hidden_size": 128, "num_layers": 3, 'num_types': 2} |
| 25 | )) |
| 26 | assert a.net.transformer.position_embeddings.weight.shape[-1] == 128 |
| 27 | |
| 28 | def test_save_and_load(): |
| 29 | args = NestedModel.get_args( |
no test coverage detected