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.