taim-gan / src /data /__init__.py
Dmmc's picture
three-model version
c8ddb9b
raw
history blame
166 Bytes
"""Dataset and custom collate function to load"""
from .collate import custom_collate
from .datasets import TextImageDataset
from .tokenizer import TAIMGANTokenizer