Skip to content

Latest commit

 

History

History
19 lines (15 loc) · 551 Bytes

File metadata and controls

19 lines (15 loc) · 551 Bytes

AttnGAN Inference via Pre-trained Model

Running inference

There are three steps involved.

  1. Create the container (optionally choose the cpu or gpu dockerfile:
    docker build -t "attngan" -f dockerfile.cpu .
    
  2. Run the container:
    docker run -it --name attngan -p 8888:8888 attngan bash
    
  3. Run the jupyter notebook.

Credits

All the code has been borrowed from https://github.com/taoxugit/AttnGAN. This repo just simplifies the evaluation api into a single Jupyter notebook instead of hosting on Azure.