Skip to content

Latest commit

 

History

History
13 lines (12 loc) · 392 Bytes

README.md

File metadata and controls

13 lines (12 loc) · 392 Bytes

ClassiFicon2021_GroupNo_04

Team

  • Puneet Mangla (Github: Puneet2000)
  • Yash Khasbarge (Github: yashkhasbage25)

Requirements

  • PyTorch/Torchvision
  • Numpy, Pandas
  • NVIDIA GPUs

Usage

  • move happy_images folder in current directory
  • run CUDA_VISIBLE_DEVICES=0,1,2,3 python3 scores.py
  • Specify the test file in Dataset in case you want to test on different test set.