Comment change
Browse files
tasks.py
CHANGED
|
@@ -48,7 +48,7 @@ def target_to_key(x, key_map, target_key):
|
|
| 48 |
return {**key_map, target_key: x}
|
| 49 |
|
| 50 |
|
| 51 |
-
# Final pretraining task used in Raffel et al., 2019 adaptated to
|
| 52 |
dataset_name = "/researchdisk/lm_training_dataset_full"
|
| 53 |
dataset_params = {"from_disk_path": dataset_name}
|
| 54 |
|
| 48 |
return {**key_map, target_key: x}
|
| 49 |
|
| 50 |
|
| 51 |
+
# Final pretraining task used in Raffel et al., 2019 adaptated to our use
|
| 52 |
dataset_name = "/researchdisk/lm_training_dataset_full"
|
| 53 |
dataset_params = {"from_disk_path": dataset_name}
|
| 54 |
|