GaParmar / img2img-turbo

One-step image-to-image with Stable Diffusion turbo: sketch2image, day2night, and more
MIT License
1.69k stars 191 forks source link

Training horses and zebras are wrong #108

Open beiqiqi opened 1 week ago

beiqiqi commented 1 week ago

Hello, I'm using train_cyclegan_turbo.py training horses and zebras is how to fix this error:

File "/home/img2img-turbo-main/src/train_cyclegan_turbo.py", line 390, in main(args) File "/home/img2img-turbo-main/src/train_cyclegan_turbo.py", line 91, in main feat_model = build_feature_extractor("clean", "cuda", use_dataparallel=False) File "/home/.conda/envs/img2img-turbo/lib/python3.10/site-packages/cleanfid/features.py", line 46, in build_feature_extractor feat_model = feature_extractor(name="torchscript_inception", resize_inside=False, device=device, use_dataparallel=use_dataparallel) File "/home/.conda/envs/img2img-turbo/lib/python3.10/site-packages/cleanfid/features.py", line 21, in feature_extractor model = InceptionV3W(path, download=True, resize_inside=resize_inside).to(device) File "/home/.conda/envs/img2img-turbo/lib/python3.10/site-packages/cleanfid/inception_torchscript.py", line 33, in init check_download_inception(fpath=path) File "/home/.conda/envs/img2img-turbo/lib/python3.10/site-packages/cleanfid/downloads_helper.py", line 21, in check_download_inception with urllib.request.urlopen(inception_url) as response, open(inception_path, 'wb') as f: PermissionError: [Errno 13] Permission denied: '/tmp/inception-2015-12-05.pt' Traceback (most recent call last): File "/home/.conda/envs/img2img-turbo/bin/accelerate", line 8, in sys.exit(main()) File "/home/.conda/envs/img2img-turbo/lib/python3.10/site-packages/accelerate/commands/accelerate_cli.py", line 48, in main args.func(args) File "/home/.conda/envs/img2img-turbo/lib/python3.10/site-packages/accelerate/commands/launch.py", line 1168, in launch_command simple_launcher(args) File "/home/.conda/envs/img2img-turbo/lib/python3.10/site-packages/accelerate/commands/launch.py", line 763, in simple_launcher raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)

I added :export TMPDIR=/home/tmp before the training command and it didn't solve the problem either