Fine-tuning on custom dataset

#1
by SaraAmd - opened

For fun-tuning, the model on our won dataset, the pipeline in your git hub throws the error for the "assert os.path.isfile(path), "Checkpoint {} not found!".format(path)
AssertionError: Checkpoint ./model/DiT-base not found!" while the dit-base-224-p16-500k-62d53a.pth file is avalable in ./model/DiT-base by downloading it from your git. Has anyone worked with this model to fune-tune it on the custom dataset?

Sign up or log in to comment