My puppy Scout was like that when we first got him. He was 4 months at the time.

We put him in bed with us and put a choker collar on him (sorry if that offends anyone, it was small and no spikes of course). I held on to the collar myself so he couldn't get away and laid him down beside me on the bed in complete darkness. He'd start to move, and feel it tighten, and he wouldn't go any further. He never actually "choked" himself on it. At first, it would take maybe an hour to get him to sleep. Within a week, he went right to sleep on his own, and we had no need for using the choker chain since. Now that he's a big boy (lol -- 6 months), he knows lights out means sleep, and he sleeps on a bean bag chair beside the bed.

Since your puppy is so small, maybe you could just hold him beside you and not let him move.

I'd say spend a few days at home with him all day long. Don't let him sleep during the day -- keep him up playing, whatever. Then by night time, he should sleep soundly. After forcing him to readjust his schedule for a few days, he should naturally readjust it. And always exercise him before bedtime. Maybe even have a wind-down ritual? We sit in the living room with the lights out and watch tv for at least 30 minutes before bed time, and the dogs wind down then. We usually have to wake them up to relocate them to the bedroom (where we can hear them if they need to go out).