Merge pull request #162 from marcbone/fix-cpu-option-in-cloth-example

Fix cpu option in example cloth_on_rigid
Genesis-Embodied-AI · Dec 21, 2024 · e72e146 · e72e146
2 parents 24b79cb + 7f52f40
commit e72e146
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(