End of Week 8 - 27/07/2020
What did you do this week?
Picking up from last week, I implemented dynamic loading of all the pre-trained torchvision models for image classification. This method creates the ``Model`` classes on the fly using type(), sys.modules and setattr() and loads the weights only for the model that is being used! I discussed with my mentor on more computer vision operations and examples that will be a great addition to DFFML.
What is coming up next?
After having a discussion with my mentor, we decided that we will work on exposing the optimization and loss function options available in PyTorch and layer support for making custom neural networks in a separate pull request next!
Did you get stuck anywhere?
No, this week went by smoothly!
Thank you for reading!
from Planet Python
via read more
No comments:
Post a Comment