Blank Images using Pyramid model
3DTOPO opened this issue · comments
I am attempting to train a model using the default pyramid model with this command (also tried using all defaults):
th train.lua -style_weight 5 -tv_weight 0.000085 -style_size 512 -model pyramid -image_size 512 -num_iterations 50000 -batch_size 1 -data ../coco2014 -style_image ../candy_512.jpg
When I try to generate an image with it, I am getting black output, even though the input image's dimension is a multiple of 32:
th test.lua -image_size 1024 -input_image ../Input_square1568.jpg -model_t7 pyramid/model_20000.t7 -save_path stylized.jpg
The issue occurs if I don't use the image_size flag as well. The weird thing is, it seemed to work a couple times, but now all I get is black output. Using johnson's model does not create the issue, but skip_unpool model also creates blank output.
I have tried creating smaller images to 256, 512 etc. but they are all blank.
I am just trying to figure out how to train the model to achieve results similar to the default model. As others have posted, it doesn't seem possible with the current code. It seems like others have experienced the black output as well. Please advise!
Actually, using the skip_unpool model causes a core dump:
THCudaCheck FAIL file=/home/prime/torch/extra/cutorch/lib/THC/generic/THCStorage.c line=147 error=77 : an illegal memory access was encountered
Segmentation fault (core dumped)
I'll take a look tomorrow.
Hi @DmitryUlyanov would it be helpful for me to upload the models?