and this is the error I got:
TypeError                                 Traceback (most recent call last)
<ipython-input-9-2a363c197ef5> in <module>()
     12         imgs = list(img.to(device) for img in imgs)
     13         annotations = [{k: v.to(device) for k, v in t.items()} for t in annotations]
---> 14         loss_dict = model(imgs, annotations)
     15         losses = sum(loss for loss in loss_dict.values())
     16 

/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py in _call_impl(self, *input, **kwargs)
    720             result = self._slow_forward(*input, **kwargs)
    721         else:
--> 722             result = self.forward(*input, **kwargs)
    723         for hook in itertools.chain(
    724                 _global_forward_hooks.values(),

TypeError: forward() takes 2 positional arguments but 3 were given