code like this:
self.bert = torch.jit.trace(bert_model,(input_ids,attention_mask))
I want to set output_hidden_states=True,return_dict=True, but forward function of BertModel has other parameter, so I must feed all parameters ?
torch version is 1.10
2.0 version seems to have example_kwarg_inputs to achieve