adobe-research/custom-diffusion

How to train the model withot using the regularized dataset

Aditi0912dec opened this issue · 2 comments

Can you please let me know how to train the model without using the regularized dataset.
In train.py there is an argument --repeat, will only enabling that work.
Thank you.

Hi,
You would also need to remove the reg_datapath and reg_caption related arguments and adding repeat:100 to the data.params.train.params in the config file. Thus the training command for a single concept would look like this along with the changes in the finetune_addtoken.yaml config file:

python -u train.py
--base configs/custom-diffusion/finetune_addtoken.yaml
-t --gpus 0,1
--resume-from-checkpoint-custom <pretrained-model.ckpt>
--caption " cat"
--datapath <cat_sample_path>
--modifier_token ""
--name "cat-sdv4"

There would be similar changes for the multi-concept case as well. Let me know if this resolves your issue.
Thanks.