Added pred and path_weights to GPU, testing...