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 來與此資料集進行互動。