Spaces:
Sleeping
Sleeping
Commit ·
5f4bf6e
1
Parent(s): ee55dd5
update
Browse files
examples/tutorials/by_deepspeed/step_2_train_model.py
CHANGED
|
@@ -37,12 +37,14 @@ def get_args():
|
|
| 37 |
parser.add_argument("--dataset_split", default=None, type=str),
|
| 38 |
parser.add_argument(
|
| 39 |
"--dataset_cache_dir",
|
| 40 |
-
default=(project_path / "hub_datasets").as_posix(),
|
|
|
|
| 41 |
type=str
|
| 42 |
),
|
| 43 |
parser.add_argument(
|
| 44 |
"--model_cache_dir",
|
| 45 |
-
default=(project_path / "hub_models").as_posix(),
|
|
|
|
| 46 |
type=str
|
| 47 |
),
|
| 48 |
parser.add_argument("--dataset_streaming", default=None, type=str),
|
|
|
|
| 37 |
parser.add_argument("--dataset_split", default=None, type=str),
|
| 38 |
parser.add_argument(
|
| 39 |
"--dataset_cache_dir",
|
| 40 |
+
# default=(project_path / "hub_datasets").as_posix(),
|
| 41 |
+
default="/root/autodl-tmp/OpenMiniMind/hub_datasets",
|
| 42 |
type=str
|
| 43 |
),
|
| 44 |
parser.add_argument(
|
| 45 |
"--model_cache_dir",
|
| 46 |
+
# default=(project_path / "hub_models").as_posix(),
|
| 47 |
+
default="/root/autodl-tmp/OpenMiniMind/hub_models",
|
| 48 |
type=str
|
| 49 |
),
|
| 50 |
parser.add_argument("--dataset_streaming", default=None, type=str),
|