wty-ustc / HairCLIP

[CVPR 2022] HairCLIP: Design Your Hair by Text and Reference Image
GNU Lesser General Public License v2.1
541 stars 68 forks source link

Error while training the model on my dataset. #46

Open ArsalanAli915 opened 1 year ago

ArsalanAli915 commented 1 year ago

s3

This is the command I am using. %%shell eval "$(conda shell.bash hook)" conda activate myenviroment python scripts/train.py \ --exp_dir=/content/outss \ --hairstyle_description="hairstyle_list.txt" \ --color_description=black,brown,yellow \ --checkpoint_path=/content/drive/MyDrive/data/hairclip.pt \ --ir_se50_weights=/content/drive/MyDrive/data/model_ir_se50.pth \ --latents_train_path=/content/drive/MyDrive/data/trainlatent/latents.pt \ --latents_test_path=/content/drive/MyDrive/data/testlatent/latents1.pt \ --hairstyle_ref_img_train_path=/content/inversions \ --hairstyle_ref_img_test_path=/content/test/inversions \ --color_ref_img_train_path=/content/inversions \ --color_ref_img_test_path=/content/test/inversions \ --color_ref_img_in_domain_path=/content/inversions \ --hairstyle_manipulation_prob=0.5 \ --color_manipulation_prob=0.2 \ --both_manipulation_prob=0.27 \ --hairstyle_text_manipulation_prob=0.5 \ --color_text_manipulation_prob=0 \ --color_in_domain_ref_manipulation_prob=0.25 \