|
OpenNN
Open-source neural networks library
|
Sampling parameters that control how the next token is drawn from the model output distribution. More...
#include <transformer_decoder.h>
Public Attributes | |
| float | temperature = 1.0f |
| Index | top_k = 0 |
| float | top_p = 1.0f |
| float | repetition_penalty = 1.0f |
| Index | maximum_tokens = 0 |
Sampling parameters that control how the next token is drawn from the model output distribution.
| Index opennn::TransformerDecoder::SamplingConfig::maximum_tokens = 0 |
| float opennn::TransformerDecoder::SamplingConfig::repetition_penalty = 1.0f |
| float opennn::TransformerDecoder::SamplingConfig::temperature = 1.0f |
| Index opennn::TransformerDecoder::SamplingConfig::top_k = 0 |
| float opennn::TransformerDecoder::SamplingConfig::top_p = 1.0f |