Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to train the model withot using the regularized dataset #69

Open
Aditi0912dec opened this issue Jul 20, 2023 · 2 comments
Open

How to train the model withot using the regularized dataset #69

Aditi0912dec opened this issue Jul 20, 2023 · 2 comments

Comments

@Aditi0912dec
Copy link

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.

@nupurkmr9
Copy link
Collaborator

nupurkmr9 commented Jul 21, 2023

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.

@Aditi0912dec
Copy link
Author

Aditi0912dec commented Jul 28, 2023 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants