fix cpu arg in cloth_on_rigid.py example

Genesis-Embodied-AI · Dec 20, 2024 · 7f52f40 · 7f52f40
1 parent be62afe
commit 7f52f40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/coupling/cloth_on_rigid.py b/examples/coupling/cloth_on_rigid.py
@@ -12,7 +12,7 @@ def main():
     args = parser.parse_args()
 
     ########################## init ##########################
-    gs.init(seed=0, precision="32", logging_level="debug")
+    gs.init(seed=0, precision="32", logging_level="debug", backend=gs.cpu if args.cpu else gs.gpu)
 
     scene = gs.Scene(
         sim_options=gs.options.SimOptions(