RTX 3080 Ti / Masked Training on CPU ?

[H4xl0r]

THIS IS NOT TECH SUPPORT FOR NEWBIE FAKERS POST ONLY ISSUES RELATED TO BUGS OR CODE

Expected behavior

Starting masked training (head) on the RTX Build.

Actual behavior

Should run on GPU but got heavy load on CPU. And training runs slow.

Steps to reproduce

.....

Other relevant information

• Command lined used (if not specified in steps to reproduce): main.py ...
• Operating system and version: Windows10
• Python version: 3.7.9

What drivers are suggested to use with the RTX3080ti? What could be the problem ?

[Ozamatheus]

Same here, it was working faster with GTX 1650 4gb, 2 or 3 iterations and stop for a second, even with resolution 128 and batch 8

[joolstorrentecalo]

Did you ever find the answer? If so, would you mind sharing it and closing this issue?