Basically, it seemed like you had to go back to the main room just because you messed up a single rock push after 4 puzzles in a row. There didn't seem to be any checkpoints other than that.