/IE4483_assignment

Primary LanguagePython

IE4483 Assignment

Cat & Dog

  1. create conda env from resnet_env.yml, conda env create --name resnet_env --file=resnet_env.yml
  2. Train cat & dog : python resnet50.py or python resnet101.py, trained weights will be saved in ./weights
  3. Create submission : python submission.py

CIFAR-10

  1. Train using CIFAR-10 dataset : python resnet50_cifar10.py or python resnet101_cifar10.py

All the results (accuracy, loss) will be saved in result folder