Spaces:

Rolv-Arild
/

next-goal-predictor

Sleeping

Rolv-Arild commited on Jun 30

Commit

3d1042f

verified ·

1 Parent(s): eb781a5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -64,7 +64,7 @@ def infer(model, replay_file,
     goal_diff = serialized_scoreboards[:, SB_BLUE_SCORE] - serialized_scoreboards[:, SB_ORANGE_SCORE]
     goal_diff_diff = goal_diff.diff(prepend=torch.Tensor([0]))
-    bs = 512
     predictions = []
     it = trange(len(serialized_states), desc="Running model")
     for i in range(0, len(serialized_states), bs):

     goal_diff = serialized_scoreboards[:, SB_BLUE_SCORE] - serialized_scoreboards[:, SB_ORANGE_SCORE]
     goal_diff_diff = goal_diff.diff(prepend=torch.Tensor([0]))
+    bs = 900
     predictions = []
     it = trange(len(serialized_states), desc="Running model")
     for i in range(0, len(serialized_states), bs):