Skip to content

Terminates after 2 iterations, no output #179

@cashpipeplusplus

Description

@cashpipeplusplus

With lbfgs, I'm getting no output image and termination after just 2 iterations.

My command-line is:

th neural_style.lua -style_image examples/inputs/starry_night.jpg -content_image examples/inputs/hoovertowernight.jpg -gpu -1 -print_iter 1

Here's the console output:

[libprotobuf WARNING google/protobuf/io/coded_stream.cc:505] Reading dangerously large protocol message.  If the message turns out to be larger than 1073741824 bytes, parsing will be halted for security reasons.  To increase the limit (or to disable these warnings), see CodedInputStream::SetTotalBytesLimit() in google/protobuf/io/coded_stream.h.
[libprotobuf WARNING google/protobuf/io/coded_stream.cc:78] The total number of bytes read was 574671192
Successfully loaded models/VGG_ILSVRC_19_layers.caffemodel
conv1_1: 64 3 3 3
conv1_2: 64 64 3 3
conv2_1: 128 64 3 3
conv2_2: 128 128 3 3
conv3_1: 256 128 3 3
conv3_2: 256 256 3 3
conv3_3: 256 256 3 3
conv3_4: 256 256 3 3
conv4_1: 512 256 3 3
conv4_2: 512 512 3 3
conv4_3: 512 512 3 3
conv4_4: 512 512 3 3
conv5_1: 512 512 3 3
conv5_2: 512 512 3 3
conv5_3: 512 512 3 3
conv5_4: 512 512 3 3
fc6: 1 1 25088 4096
fc7: 1 1 4096 4096
fc8: 1 1 4096 1000
Running optimization with L-BFGS
Iteration 1 / 1000
  Total loss: 0.000000  
<optim.lbfgs>   creating recyclable direction/step/history buffers
Iteration 2 / 1000
  Total loss: 0.000000  
<optim.lbfgs>   function value changing less than tolX  

If I switch to adam, I get all 1000 iterations, each with Total loss: 0.000000 afterward and this image as output:
image

Am I doing something wrong?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions