Ohh ok i understand the lua interface now.
How could I translate it into pytorch then ? With a loss fonction ?
If I understand it well there is the attributes gradinput in modules for torch lua, but I don’t understand how I could access it with pytorch.
Thank you for your answer taken on your time, I am a beginner concerning the implementation with pytorch