CNN model for prediction using Pytorch library

-create CNN model with 3 layers for prediction using Pytorch library
- write code to repeat the training process several times and calculate the average of error metrics
- use TensorBoard to create visualizations of the model and the training process