From 12a0cb6286c985b7c04ff1e77aedccb5a583ac4d Mon Sep 17 00:00:00 2001 From: Joel Holdbrooks Date: Sun, 29 Jan 2023 17:58:42 -0800 Subject: [PATCH] Update documentation --- data/every_dream.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/data/every_dream.py b/data/every_dream.py index 0d0cd41..e21d639 100644 --- a/data/every_dream.py +++ b/data/every_dream.py @@ -25,14 +25,11 @@ import torch.nn.functional as F class EveryDreamBatch(Dataset): """ - data_root: root path of all your training images, will be recursively searched for images - repeats: how many times to repeat each image in the dataset - flip_p: probability of flipping the image horizontally + data_loader: `DataLoaderMultiAspect` object debug_level: 0=none, 1=print drops due to unfilled batches on aspect ratio buckets, 2=debug info per image, 3=save crops to disk for inspection - batch_size: how many images to return in a batch conditional_dropout: probability of dropping the caption for a given image - resolution: max resolution (relative to square) - jitter: number of pixels to jitter the crop by, only for non-square images + crop_jitter: number of pixels to jitter the crop by, only for non-square images + seed: random seed """ def __init__(self, data_loader: DataLoaderMultiAspect,