victoresque / pytorch-template

PyTorch deep learning projects made easy.
MIT License
4.75k stars 1.09k forks source link

Adding task 1 branch #101

Closed Abhi-Tiw1 closed 2 years ago

Abhi-Tiw1 commented 2 years ago

Task 1 simply includes modifying the mode to have different components of the network tweaked, this includes batch norm instead of dropout, changing kernal sizes and changing pooling layers..