Dalle2_pytorch_project

Implementation of Dalle2 Pytorch

Download Coco & VizWiz Dataset

1. Run coco_dataset.py
2. Run vizwiz_dataset.py
3. unzip all the dataset using winrar

Training

1. Run "pip install -r requirements.txt" on terminal
2. Set the parameters in train_dalle2.ipynb or train_dalle2.py (choose one)
3. Run train_dalle2.ipynb or train_dalle2.py (choose one)

Note: If you want to train the model in several datasets, just run the train_dalle2.ipynb or train_dalle2.py several times with different train data path in each run.

Testing

1. Run "pip install -r requirements.txt" on terminal (Skip if you have run it before)
2. Set the parameters in test_dalle2.ipynb or test_dalle2.py (choose one)
3. Run test_dalle2.ipynb or test_dalle2.py (choose one)