Merge pull request #10 from yaacov/tweak

random tweak
RedHat-Israel · Jan 25, 2024 · f39e726 · f39e726
2 parents db83128 + 4e215d2
commit f39e726
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/checkpoints/driver.pth b/checkpoints/driver.pth
diff --git a/train.py b/train.py
@@ -142,7 +142,7 @@ def generate_batch(batch_size):
     targets = []
 
     for _ in range(batch_size):
-        car_x = random.choice([0, 1, 2, 3, 4, 5])
+        car_x = random.randint(0, 5)
         array = generate_obstacle_array()
         correct_output = driver_simulator(array, car_x)