I have added classification and regression head to resnet50 pretrained model. But while training after the fine-tuning getting below error message:
It seems the input is a
list while a tensor is expected.
Post a minimal and executable code snippet instead of screenshots for better debuggability.
Thanks for the input. I converted input list to tensor. Its working now