Skip to content

Latest commit

 

History

History
21 lines (17 loc) · 716 Bytes

README.md

File metadata and controls

21 lines (17 loc) · 716 Bytes

Self Attention Generative Adversarial Networks

Procedure to run the project:

  • Install the Celeb-A dataset

  • Celeb-A Faces dataset can be downloaded at the linked site, or in Google Drive. The dataset will download as a file named img_align_celeba.zip

  • Extract the dataset into data/celeba

  • Install Pytorch, NumPy, Matplotlib

  • Install Pytorch Ignite using

    pip:

      pip install pytorch-ignite
    

    conda:

      conda install ignite -c pytorch
    
  • CUDA Device(GPU) must be present to run the IPython notebook.

  • Run all cells of the IPython Notebook.