AI-DungeonMaster icon indicating copy to clipboard operation
AI-DungeonMaster copied to clipboard

Pass temperature and top_k to sampler in generator.py

Open kinoc opened this issue 5 years ago • 0 comments

In generator.py I changed:

    _self.output = sample.sample_sequence(
        hparams=hparams, length=length,
        context=self.context,
        batch_size=batch_size,
        )_

to self.output = sample.sample_sequence( hparams=hparams, length=length, context=self.context, batch_size=batch_size, temperature=temperature, top_k=top_k )

I am trying some finetuned models and those values make a world of difference in coherence and quality.

kinoc avatar Apr 21 '19 04:04 kinoc