Mirrors
/
EveryDream-trainer
mirror of https://github.com/victorchall/EveryDream-trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix dl test script

This commit is contained in:
Victor Hall 2022-12-17 02:12:38 -05:00
parent 6a22fa7594
commit 3d4bc5d8de
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_dl.py
View File

@ -6,8 +6,8 @@ import argparse
import ldm.data.data_loader as dl import ldm.data.data_loader as dl
def main(data_root, batch_size): def main(data_root, batch_size, resolution):
data_loader = dl.DataLoaderMultiAspect(data_root=data_root, batch_size=batch_size, debug_level=1, resolution=512) data_loader = dl.DataLoaderMultiAspect(data_root=data_root, batch_size=batch_size, resolution=resolution, debug_level=1)
image_caption_pairs = data_loader.get_all_images() image_caption_pairs = data_loader.get_all_images()