Open ArsalanAli915 opened 1 year ago
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 \
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 \