Adversarial-Deep-Learning / code-soup

This is a collection of algorithms and approaches used in the book adversarial deep learning

MIT License

18 stars 18 forks source link

Updated ReadME.md under vision, imported torchvision models #57

Closed pronoma closed 2 years ago

pronoma commented 2 years ago

This PR adds to the model ReadME to provide information on the input sizes for various pre-trained TorchVision models.

codecov[bot] commented 2 years ago

Codecov Report

Merging #57 (f9de760) into main (d6b5cdd) will not change coverage. The diff coverage is 100.00%.

@@            Coverage Diff            @@
##              main       #57   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files            8         8           
  Lines          191       192    +1     
=========================================
+ Hits           191       192    +1

Impacted Files	Coverage Δ
code_soup/common/vision/models/__init__.py	`100.00% <100.00%> (ø)`

abheesht17 commented 2 years ago

Thank you, @pronoma! We can directly load the model using torchvision.models.inception_v3(). I don't think we need the model classes.

pronoma commented 2 years ago

Hello, I have updated a table under 'code_soup/common/vision/models/readme.md' containing the input size details on pre-trained torchvision models. I have also imported most of the torchvision models under 'code_soup/common/vision/models/init.py'.

someshsingh22 commented 2 years ago

Great!