INNER CODE UNIT · Python
nondefault_trainer_args
TIGER-AI-Lab/ImagenHub · src/imagen_hub/pipelines/custom_diffusion/custom_diffusion_src/stable-diffusion/main.py:129
def nondefault_trainer_args(opt):
parser = argparse.ArgumentParser()
parser = Trainer.add_argparse_args(parser)
args = parser.parse_args([])
return sorted(k for k in vars(args) if getattr(opt, k) != getattr(args, k))
class WrappedDataset(Dataset):
"""Wraps an arbitrary object with __len__ and __getitem__ into a pytorch dataset"""
def __init__(self, dataset):
self.data = dataset
def __len__(self):
return len(self.data)
def __getitem__(self, idx):
return self.data[idx]