# s = maze.stationary_densities(
# exit(0)
####
- masks = mazes == maze.v_empty
output_gpt = gpt(mygpt.BracketedSequence(mazes), mode=args.oneshot_input).x
output = model(output_gpt)
mazes.view(-1, task.height, task.width),
policies.view(-1, 4, task.height, task.width),
).flatten(-2)
- scores = output.flatten(-2)
+ scores = output
+ print(scores)
else:
raise ValueError(f"{args.oneshot_output=}")