Description
Traceback (most recent call last):
File "/home/zhsx/RL4LMs/scripts/training/train_text_generation.py", line 84, in
main(
File "/home/zhsx/RL4LMs/scripts/training/train_text_generation.py", line 46, in main
trainer = OnPolicyTrainer(
File "/home/zhsx/RL4LMs/rl4lms/envs/text_generation/training_utils.py", line 149, in init
self._setup()
File "/home/zhsx/RL4LMs/rl4lms/envs/text_generation/training_utils.py", line 160, in _setup
self._samples_by_split = build_datapool(
File "/home/zhsx/RL4LMs/rl4lms/envs/text_generation/training_utils.py", line 66, in build_datapool
train_datapool = _get_datapool_by_split("train")
File "/home/zhsx/RL4LMs/rl4lms/envs/text_generation/training_utils.py", line 63, in _get_datapool_by_split
dp_split = DataPoolRegistry.get(datapool_config["id"], kwargs)
File "/home/zhsx/RL4LMs/rl4lms/envs/text_generation/registry.py", line 107, in get
datapool = datapool_cls.prepare(**kwargs)
File "/home/zhsx/RL4LMs/rl4lms/data_pools/custom_text_generation_pools.py", line 183, in prepare
dataset = load_dataset("cnn_dailymail")
File "/home/zhsx/miniconda3/envs/dsp1/lib/python3.10/site-packages/datasets/load.py", line 1675, in load_dataset
builder_instance = load_dataset_builder(
File "/home/zhsx/miniconda3/envs/dsp1/lib/python3.10/site-packages/datasets/load.py", line 1452, in load_dataset_builder
dataset_module = dataset_module_factory(
File "/home/zhsx/miniconda3/envs/dsp1/lib/python3.10/site-packages/datasets/load.py", line 1177, in dataset_module_factory
raise e1 from None
File "/home/zhsx/miniconda3/envs/dsp1/lib/python3.10/site-packages/datasets/load.py", line 1156, in dataset_module_factory
return HubDatasetModuleFactoryWithoutScript(
File "/home/zhsx/miniconda3/envs/dsp1/lib/python3.10/site-packages/datasets/load.py", line 743, in init
assert self.name.count("/") == 1
AssertionError