/pytorch-vgg16

implementing VGG-16 using pytorch framework

Primary LanguageJupyter Notebook

Implementation of VGG-16 using pytorch framework

🔗 Links

Data_set_link

Google_colab_link

Installation

If you are working in jupyter-notbook

  conda create -n pytorch-basic python==3.7 -y

  conda activate pytorch-basic
  
  pip3 install torch torchvision torchaudio

If you have GPU then refer the official pytorch website for the Installation

Logo