clarification of the pytorch code ready to classify two class of images

Job ID: 32160480

Budget: $10 – $30 USD

Hello, I have a simple pytorch code and I need the following :
- I need this code to be executed on the cpu and not the gpu to
classify two categories of images instead of 6 categories by using
pycharm.
- Simplify the code further.
- summarizing the network.
- draw (plot) the artificial neural network.
- display training epoches.
- write a full report explaining the entire project with a step-by-step
by reinforcement with pictures of the implementation process.