Problem with forward step in pytorch model

Question

Traceback saying that:

* Epoch 1/20
---------------------------------------------------------------------------
TypeError                                 Trac