Converting the code to PyTorch
Problem to solve
I want PyTorch to be the library used for the development of the DL applications.
Further details
Proposal
Machine Learning Models and Training
- The models to be written in PyTorch.
- Training of PyTorch models to be executed succesfully in one GPU.
- The code to be refactored.