da03 commited on
Commit
85b44f8
·
1 Parent(s): 1d0ebde
Files changed (1) hide show
  1. main.py +4 -1
main.py CHANGED
@@ -201,8 +201,10 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
201
  previous_actions.insert(0, ("move", unnorm_coords(x, y)))
202
  prev_x = 0
203
  prev_y = 0
 
204
 
205
  if DEBUG_TEACHER_FORCING:
 
206
  # Use the predefined actions for image_81
207
  debug_actions = [
208
  'N + 0 8 5 3 : + 0 4 5 0', 'N + 0 8 7 1 : + 0 4 6 3',
@@ -215,11 +217,12 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
215
  'N + 0 9 2 7 : + 0 5 0 1'
216
  ]
217
  previous_actions = []
218
- for action in debug_actions:
219
  x, y, action_type = parse_action_string(action)
220
  previous_actions.append((action_type, (x, y)))
221
 
222
  for action_type, pos in previous_actions: #[-8:]:
 
223
  if action_type == "move":
224
  x, y = pos
225
  #norm_x = int(round(x / 256 * 1024)) #x + (1920 - 256) / 2
 
201
  previous_actions.insert(0, ("move", unnorm_coords(x, y)))
202
  prev_x = 0
203
  prev_y = 0
204
+ print ('here')
205
 
206
  if DEBUG_TEACHER_FORCING:
207
+ print ('here2')
208
  # Use the predefined actions for image_81
209
  debug_actions = [
210
  'N + 0 8 5 3 : + 0 4 5 0', 'N + 0 8 7 1 : + 0 4 6 3',
 
217
  'N + 0 9 2 7 : + 0 5 0 1'
218
  ]
219
  previous_actions = []
220
+ for action in debug_actions[-8:]:
221
  x, y, action_type = parse_action_string(action)
222
  previous_actions.append((action_type, (x, y)))
223
 
224
  for action_type, pos in previous_actions: #[-8:]:
225
+ print ('here3')
226
  if action_type == "move":
227
  x, y = pos
228
  #norm_x = int(round(x / 256 * 1024)) #x + (1920 - 256) / 2