Open ErSKS opened 5 years ago
just type
python main.py --cfg cfg/birds_3stages.yml
or whatever cfg file you want to use
@akhilvasvani does this only work with training from scratch? If I want to use one the pretrained models do I need to be loading this on an environment with an NVIDA GPU and CUDA installed?
I am trying to implement your work without using CUDA, because I don't have GPU of NVIDIA but working in gaming PC with 32GB RAM.