Spaces:
Runtime error
Runtime error
da03
commited on
Commit
·
85b44f8
1
Parent(s):
1d0ebde
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
|