Open chengkeng opened 3 years ago
you ever figure this out? having the exact same issue. it's an issue with git pretty sure, not the nst script
git clone https://github.com/gordicaleksa/pytorch-neural-style-transfer-johnson
and then follow the setup, should work
I can confirm that cloning the repo first eliminates this error.
Originally I had just put the files from the repo into the working directory manually and I too got the above error when I tried to train the network.
Hi! I tried the example and felt very cool! But when I trained my model, an error occurred in batch = [1501/20695]. The content is as follows:
(neural-style-transfer-fast) C:\MyWork\My_GAN_Work\pytorch-neural-style-transfer-fast-master>python training_script.py --style_img_name guanyu.png Using 165560 datapoints (41390 batches) (MS COCO images) for transformer network training. Learning the style of guanyu.png style image.
Hyperparams: content_weight=1.0, style_weight=400000.0 and tv_weight=0
Logging to console every 500 batches. Saving checkpoint models every 2000 batches. Tensorboard enabled. Run "tensorboard --logdir=runs --samples_per_plugin images=50" from your conda env Open http://localhost:6006/ in your browser and you're ready to use tensorboard!
I did not find a solution on Google. Is there any way to make it work? I am looking forward to the effect after training! Thank you!