Hello,
Could you please let me know the following:
Are you using the Your phone app?
What is your current setting for screen timeout? You can find this in settings > display
Is this happening on battery, when plugged in, or both?
Could you let me know the build number for your device? You can find this under settings > about