@yessyw
We have a end to end example here:
https://pytorch.org/cppdocs/frontend.html 17 (scroll down to end to end example)
Yes, you might need optimizer, we have several optimizer available. check here:
‘https://github.com/pytorch/pytorch/issues/28440 3’
And yes you can add your own optimizer, they should be the same as how you add forward and backward, same python binding, only the input and output are different.