Error running training.py
Alby-Thomas opened this issue · comments
Dear wang, I really appreciate the work you have done.I run pretrain.py and got model in pretrain folder but got an error when i run train.py. The error is like this
Traceback (most recent call last):
File "train.py", line 206, in
train(args)
File "train.py", line 193, in train
str(total_iter)+'_face_result.jpg', 4)
File "/content/drive/My Drive/cartoon/wbc_100/White-box-Cartoonization/train_code/utils.py", line 160, in write_batch_image
image[k] = (image[k]+1) * 127.5
IndexError: index 1 is out of bounds for axis 0 with size 1]
it seems the image[k] is empty, maybe you changed batch size?
These few lines of code are just saving results during training, you can just comment them and training process can work fine without them.
Thanku Wang for your reply. I will try it out and let you know.
Yes that happened because I changed the batch size. Now training error changed.