Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
sdtblck authored Jan 9, 2021
1 parent fe66e34 commit b1b6517
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,8 @@ tensorboard --logdir your_model_dir

## Dataset Creation [DALL-E]

Once the VAE is pretrained, you can move on to DALL-E
Once the VAE is pretrained, you can move on to DALL-E.

Currently we are training on a dummy dataset. A public, large-scale dataset for DALL-E is in the works. In the meantime, to generate some dummy data, run:

```bash
Expand All @@ -57,7 +58,7 @@ python data/create_tfrecords.py

This should download CIFAR-10, and generate some random captions to act as text inputs.

If you want to create your own datasets, they should be formatted in a folder, with a jsonl file in the root folder containing caption data and paths to the respective images, as follows:
Custom datasets should be formatted in a folder, with a jsonl file in the root folder containing caption data and paths to the respective images, as follows:

```
Folder structure:
Expand Down

0 comments on commit b1b6517

Please sign in to comment.