imwithye commited on
Commit
12bbb60
·
1 Parent(s): 67ab4fa

use smaller

Browse files
Files changed (1) hide show
  1. rlcube/main.py +1 -1
rlcube/main.py CHANGED
@@ -25,7 +25,7 @@ def solve(body: StateArgs):
25
 
26
  env = Cube2Env()
27
  env.reset(state=np.array(state, dtype=np.int8))
28
- tree = MonteCarloTree(env.obs(), max_simulations=2048)
29
  if tree.is_solved:
30
  return {"steps": [action for _, action in tree.solved_path]}
31
  raise HTTPException(status_code=422, detail="Unable to solve the cube")
 
25
 
26
  env = Cube2Env()
27
  env.reset(state=np.array(state, dtype=np.int8))
28
+ tree = MonteCarloTree(env.obs(), max_simulations=256)
29
  if tree.is_solved:
30
  return {"steps": [action for _, action in tree.solved_path]}
31
  raise HTTPException(status_code=422, detail="Unable to solve the cube")