Skip to content

Commit 8a01f68

Browse files
authored
readd cuda by default (#40)
that was lost in debugging
1 parent aa542ef commit 8a01f68

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

ibpc_py/src/ibpc/ibpc.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -155,8 +155,6 @@ def main():
155155
fetch_parser.add_argument("--dataset-path", default=".")
156156

157157
extension_manager = RockerExtensionManager()
158-
default_args = {"cuda": True, "network": "host"}
159-
# extension_manager.extend_cli_parser(test_parser, default_args)
160158

161159
args = main_parser.parse_args()
162160
args_dict = vars(args)
@@ -227,6 +225,7 @@ def run_instance(dig_instance, args):
227225
args_dict["name"] = "bpc_estimator"
228226
args_dict["network"] = "host"
229227
args_dict["extension_blacklist"] = ({},)
228+
args_dict["cuda"] = True
230229

231230
# Confirm dataset directory is absolute
232231
args_dict["dataset_directory"] = os.path.abspath(args_dict["dataset_directory"])

0 commit comments

Comments
 (0)