Skip to content

Commit

Permalink
be explicit about running forward
Browse files Browse the repository at this point in the history
Signed-off-by: yaacov <[email protected]>
  • Loading branch information
yaacov committed Jul 1, 2024
1 parent f39e726 commit 76d5cc7
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mydriver.py
Original file line number Diff line number Diff line change
Expand Up @@ -90,8 +90,8 @@ def drive(world):
view = build_lane_view(world)
input_tensor = view_to_inputs(view, world.car.x).unsqueeze(0)

# Use neural network model to get the outputs tensor
output = model(input_tensor)
# Forward propagte the inputs in the neural network model to get the outputs tensor
output = model.forward(input_tensor)

# Convert the output tensor into a real world response
action = outputs_to_action(output)
Expand Down

0 comments on commit 76d5cc7

Please sign in to comment.