We have Weimaraners that are crate trained. Big Dogs. They love their "night-nights". They go in for day naps or whenever they want, they are fed in their crates, and stay there for 1 1/2 hrs. after meals because they are suseptable to bloat. They have lots of blankets in their crates, & many toys. My daughter used to give them the bong toy with a little peanut butter treat in it in the crate when they were young. It takes them awhile to get the peanut butter out of it & is fun for them.
They go in their crate when we aren't home for their protection even tho they are highly trained. When they anticipate that we are leaving, they just go get in their crates. It also gives us piece of mind that they won't hurt themselves while we are gone. Sometimes we give them a little treat in their crates as we leave.
Their crates are very big so they can stand, lay down, move around, lay sideways, what ever they want while in them. They don't sleep in them at night but could & would if we wanted them to.
Crates are a good thing & it probably won't take long for your puppy to get used to it. It shouldn't be a punishiment!
It's their safe place when they feel the need to be safe.
Good luck & make the crate a good place that he/she is happy to be in.