Convolutional Neural Networks - Weekly Check-in 8

Published: 07/20/2020

End of Week 7  - 20/07/2020

What did you do this week?

This week I started working on adding pre-trained Convolutional Neural Networks using the PyTorch Library! I have currently added VGG16 and ResNet18 pre-trained Models and will add more after further discussion with my mentor. 

What is coming up next?

I will work on exposing the optimizing options available in the PyTorch library through DFFML to get the best results using the added models .

Did you get stuck anywhere?

I got a little confused on whether to split the train data into training and validation sets or not. Eventually I went with providing a splitting argument to split train data into validation and training data sets.

Thank you for reading!