error running style.py

[YvesBB]

I' m getting an error when i want to train with style.py , it runs perfectly if i evaluate a style transfer with evaluate.py

python style.py --style images/emoticon.jpg --checkpoint-dir checkpoint/ --content-weight 1.5e1 --checkpoint-iterations 1000 --batch-size 20

CUDA Version: 11.2
Python 3.9.4
Tensorflow 2.5.0-rc0
GeForce gtx 1060

pb.txt

[keremkurtulus]

I fixed it with #10 (comment)

[mmethot]

I had to specify a smaller batch-size. I removed the --batch-size command line option entirely from my command to let it take the default, which is 4 and which worked.