Fix step count

PySymGym · Feb 21, 2025 · c9728c8 · c9728c8
1 parent 17bc8ea
commit c9728c8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/VSharp.ML.GameServer.Runner/Main.fs b/VSharp.ML.GameServer.Runner/Main.fs
@@ -350,6 +350,8 @@ let runTrainingSendModelMode
     (port: int)
     =
     printfn $"Run infer on {gameMap.MapName} have started."
+    let stepsToStart = gameMap.StepsToStart
+    let stepsToPlay = gameMap.StepsToPlay
 
     let aiTrainingOptions =
         {
@@ -363,8 +365,8 @@ let runTrainingSendModelMode
 
                     mapName = gameMap.MapName
                 }
-            stepsToSwitchToAI = gameMap.StepsToStart
-            stepsToPlay = gameMap.StepsToPlay
+            stepsToSwitchToAI = stepsToStart
+            stepsToPlay = stepsToPlay
             oracle = None
         }
 
@@ -391,6 +393,7 @@ let runTrainingSendModelMode
             outputDirectory = outputDirectory,
             searchStrategy = SearchStrategy.AI,
             solverTimeout = SOLVER_TIMEOUT_FOR_TRAINING,
+            stepsLimit = uint (stepsToPlay + stepsToStart),
             aiOptions = (Some aiOptions |> Option.defaultValue Unchecked.defaultof<_>),
             pathToModel = pathToModel,
             useGPU = useGPU,