End of Week 6 - 13/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!