Text-classification out of memory for GPU training

[zewditu]

System Information (please complete the following information):

• Model Builder Version (available in Manage Extensions dialog):
• Visual Studio Version

Describe the bug

• On which step of the process did you run into an issue:
• Clear description of the problem:

To Reproduce
Steps to reproduce the behavior:

1. Go to '...'
2. Click on '....'
3. Scroll down to '....'
4. See error

Expected behavior
A clear and concise description of what you expected to happen.

Screenshots

Additional context
Add any other context about the problem here.

[luisquintanilla]

Being worked on as part of dotnet/docs#32011