make_gsm8k_env

class torchrl.envs.llm.make_gsm8k_env(dataset: str = 'gsm8k', num_envs: int = 1, repeats: int | None = None, batch_size_dl: int = 1, seed: int | None = None, group_repeats: bool = False, tokenizer: transformers.PretrainedTokenizer | None = None)

构建基于 LLMEnv 的 GSM8K 环境的生成器。

注意

建议使用 torchrl.envs.llm.GSM8KEnv 来与此数据集进行交互。