The code was implemented using Python 3.6.0 and trained on one NIVIDIA GeForce GTX TITAN X GPU.
We provide the pre-trained models on CUB in save/ and test script in test_on_CUB.py.
The testing command as:
python test_on_CUB.py| Name | Name | Last commit date | ||
|---|---|---|---|---|
The code was implemented using Python 3.6.0 and trained on one NIVIDIA GeForce GTX TITAN X GPU.
We provide the pre-trained models on CUB in save/ and test script in test_on_CUB.py.
The testing command as:
python test_on_CUB.py