- In order to run the code Docker, Docker-Compose and NVIDIA-Docker are required. The code has been tested on a NVIDIA RTX 2080 Ti (8 images per minute)
- Download
pytorch_model.bin
from here and place it in the root folder.
- In the
docker
folder rundocker-compose up -d
- Access the container
docker exec -it main bash
- Start the script
python main.py
.
Now your GPU will generate pokemons forever :)