File tree 1 file changed +3
-1
lines changed
1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -64,6 +64,7 @@ def __init__(
64
64
rdzv_backend : str = None ,
65
65
scheduler_args : Optional [Dict [str , str ]] = None ,
66
66
image : Optional [str ] = None ,
67
+ workspace : Optional [str ] = f"file://{ Path .cwd ()} " ,
67
68
):
68
69
if bool (script ) == bool (m ): # logical XOR
69
70
raise ValueError (
@@ -87,6 +88,7 @@ def __init__(
87
88
scheduler_args if scheduler_args is not None else dict ()
88
89
)
89
90
self .image = image
91
+ self .workspace = workspace
90
92
91
93
def _dry_run (self , cluster : "Cluster" ):
92
94
j = f"{ cluster .config .max_worker } x{ max (cluster .config .gpu , 1 )} " # # of proc. = # of gpus
@@ -113,7 +115,7 @@ def _dry_run(self, cluster: "Cluster"):
113
115
),
114
116
scheduler = cluster .torchx_scheduler ,
115
117
cfg = cluster .torchx_config (** self .scheduler_args ),
116
- workspace = f"file:// { Path . cwd () } " ,
118
+ workspace = self . workspace ,
117
119
)
118
120
119
121
def _missing_spec (self , spec : str ):
You can’t perform that action at this time.
0 commit comments