yuntian-deng commited on
Commit
03c7138
·
1 Parent(s): 7b868a7

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -358,7 +358,7 @@ async def websocket_endpoint(websocket: WebSocket):
358
  next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
359
  # Load and append the corresponding ground truth image instead of model output
360
  print ('here4', len(previous_frames))
361
- if DEBUG_TEACHER_FORCING:
362
  img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
363
  previous_frames.append(img)
364
  else:
 
358
  next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
359
  # Load and append the corresponding ground truth image instead of model output
360
  print ('here4', len(previous_frames))
361
+ if False and DEBUG_TEACHER_FORCING:
362
  img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
363
  previous_frames.append(img)
364
  else: