We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aa542ef commit 8a01f68Copy full SHA for 8a01f68
ibpc_py/src/ibpc/ibpc.py
@@ -155,8 +155,6 @@ def main():
155
fetch_parser.add_argument("--dataset-path", default=".")
156
157
extension_manager = RockerExtensionManager()
158
- default_args = {"cuda": True, "network": "host"}
159
- # extension_manager.extend_cli_parser(test_parser, default_args)
160
161
args = main_parser.parse_args()
162
args_dict = vars(args)
@@ -227,6 +225,7 @@ def run_instance(dig_instance, args):
227
225
args_dict["name"] = "bpc_estimator"
228
226
args_dict["network"] = "host"
229
args_dict["extension_blacklist"] = ({},)
+ args_dict["cuda"] = True
230
231
# Confirm dataset directory is absolute
232
args_dict["dataset_directory"] = os.path.abspath(args_dict["dataset_directory"])
0 commit comments