.

main.py

@@ -213,9 +213,9 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
             #norm_y = int(round(y / 256 * 640)) #y + (1080 - 256) / 2
             norm_x = x + (1920 - 512) / 2
             norm_y = y + (1080 - 512) / 2
-            #if DEBUG:
-            #    norm_x = x
-            #    norm_y = y
+            if DEBUG_TEACHER_FORCING:
+                norm_x = x
+                norm_y = y
             #action_descriptions.append(f"{(norm_x-prev_x):.0f}~{(norm_y-prev_y):.0f}")
             #action_descriptions.append(format_action(f'{norm_x-prev_x:.0f}~{norm_y-prev_y:.0f}', x==0 and y==0))
             action_descriptions.append(format_action(f'{norm_x:.0f}~{norm_y:.0f}', x==0 and y==0))
@@ -225,8 +225,9 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
             x, y = pos
             #norm_x = int(round(x / 256 * 1024)) #x + (1920 - 256) / 2
             #norm_y = int(round(y / 256 * 640)) #y + (1080 - 256) / 2
-            norm_x = x + (1920 - 512) / 2
-            norm_y = y + (1080 - 512) / 2
+            if DEBUG_TEACHER_FORCING:
+                norm_x = x + (1920 - 512) / 2
+                norm_y = y + (1080 - 512) / 2
             #if DEBUG:
             #    norm_x = x
             #    norm_y = y