The problem is that the passed activation map contains more features than you’ve specified in the linear layer, which should accept this input.



 Muhammad_Izaz:

it has been running still now for three hours and did not even show a single output whats the problem


When should what kind of output be printed?
If you are expecting an output in each iteration, your code might hang.
In this case, could you set num_workers=0 in your DataLoader and rerun the script?