I got a little help from stack and figured out the solution.  I was using a custom dataset, which i thought also meant a dataloader.  They arent the same.  With the addition of adding my batch size into the dataloader and then unsqueezing my output I got to the solution.  Happy to help others if you run into this down the road.