このcommitはexprorer_msmdでmpsを実装するためのものです。

keisuke-yanagisawa · Apr 30, 2024 · ac01c5c · ac01c5c
1 parent d9bfd3f
commit ac01c5c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/exprorer_msmd b/exprorer_msmd
@@ -175,7 +175,8 @@ if __name__ == "__main__":
     # Raise EnvironmentError if GPU is not available
     gpuids = get_gpuids()
     # gpuids = [0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]
-    gpuids = ast.literal_eval(args.mps)
+    if args.mps is not None:
+        gpuids = ast.literal_eval(args.mps)
 
     ngpus = len(gpuids)
     if not setting["general"]["multiprocessing"]: