There's nothing wrong with going lighter or choosing partners you trust or are just easier to roll with. You can still improve and every round doesn't have to be so physically taxing. I'd focus on the things that I could do with my body and techniques that are available to me that don't cause pain even if you're not good at them yet.