ryankiros / neural-storyteller

A recurrent neural network for generating little stories about images
2.96k stars 541 forks source link

Fixed stochastic sampling (it was indexing a scalar). #8

Open reidsanders opened 8 years ago

reidsanders commented 8 years ago

Added stochastic sampling as a possible argument in generate.story, and a check to set beam_width to 1 if stochastic sampling is used.