Option of adding a switch between CPU/GPU training under training tab

Discussions about research, Faceswapping and things that don't fit in the other categories here.


Post Reply
User avatar
ugramund
Posts: 27
Joined: Sat Feb 20, 2021 3:56 am
Location: India
Has thanked: 25 times
Been thanked: 3 times

Option of adding a switch between CPU/GPU training under training tab

Post by ugramund »

Mr@torzdf , is it possible for you to add a CPU /GPU training switch under training tab
so that we can switch training between CPU & GPU like we can switch between different models
under training tab?

I am asking this question because right now I am training on a RealFace model with a batch 5 on
my CPU & its taking 5.2GB of RAM but that's not a problem since my Laptop has a 8GB RAM
installed.

But if someone else who has a GPU of 4GB VRAM may not be able to use the RealFace model at batch 5.

So in such cases can he benefit from CPU training or will he get faster results with smaller batch on that GPU?

In love with the the RealFace model.


User avatar
torzdf
Posts: 1433
Joined: Fri Jul 12, 2019 12:53 am
Answers: 125
Has thanked: 47 times
Been thanked: 278 times

Re: Option of adding a switch between CPU/GPU training under training tab

Post by torzdf »

Training on pretty much any old GPU will be orders of magnitude faster than training on CPU.

That said, CPU mode can be forced by excluding all the GPUs in the "Exclude GPUs" section

My word is final


Post Reply