Saving weights to gpu problem debugging.