| import argparse | |
| from datasets.flow_datasets import AnimeRun, fetch_dataloader | |
| if __name__ == '__main__': | |
| data = AnimeRun(split='test', dstype='Frame_Anime') | |
| print(len(data)) | |
| args = argparse.Namespace() | |
| args.stage = 'anime' | |
| args.dstype = 'Frame_Anime' | |
| args.image_size = [384, 768] | |
| args.batch_size = 12 | |
| loader = fetch_dataloader(args) | |
| print(len(loader)) | |