New puppies are always a trial--I forget mostly, when we're between puppies, but occasionally I still have PTSD-type flashbacks!
But they usually eventually mature and mellow out, so don't give up just yet. I even find myself missing some of those puppy, high-energy, hi-jinxes...now and then.
With luck, the trainer will be able to give you some good pointers and will know of a behaviorist who can help.