Hello, I some issues with the output_dir parameter.
In the yaml file, I changed it to a custom path.
Running training (on schnell) outputs an error about unexpected CustomImageDataset.__init__() argument - output_dir
I "fixed" that issue by patching it to the initmethod:
class CustomImageDataset(Dataset):
def __init__(self, img_dir, img_size=512, output_dir=None):
That allowed training to run normally, without errors.
However, here comes the second problem: nothing was returned to the output_dir directory.
Hello, I some issues with the
output_dirparameter. In the yaml file, I changed it to a custom path.Running training (on schnell) outputs an error about unexpected
CustomImageDataset.__init__()argument -output_dirI "fixed" that issue by patching it to the initmethod:That allowed training to run normally, without errors. However, here comes the second problem: nothing was returned to the
output_dirdirectory.