This seems like a python error; if you can you should post you train_epoch code.
I think the problem is that you’ve defined input_ids to be a non keyword argument.
For reference:
def keyword_arg_fn(first_param=1): 
    <some code> 

def non_keyword_arg_fn(first_param): 
   <some code> 

keyword arguments are params to a function with a default value. See here for more details.
The fix is just to remove forward input_ids in the function call.
outputs = model(input_ids,attention_mask)

rather than
outputs = model(input_ids=input_ids, attention_mask=attention_mask)