search

openai / guided-diffusion

MIT License
6.06k stars 807 forks source link

How to disable random x-flip? #48

Closed joandies closed 2 years ago

joandies commented 2 years ago

I want to train a dataset where I don't want horizontal flips, I don't find the arg to stop these random flips.

unixpickle commented 2 years ago

I don't think there is currently a command-line flag for it, but you could hard-code a change to this line:

https://github.com/openai/guided-diffusion/blob/main/guided_diffusion/image_datasets.py#L19