Page 1 of 1

It's it possible to use edge-TPU to train a modle?

Posted: Thu Feb 27, 2020 2:31 pm
by jhsy1209021

Google seemed to have been released a tpu which have been working on their server for picture searching. Is it possible to use their Dev board(edge-TPU) to train the faceswap modle?


Re: It's it possible to use edge-TPU to train a modle?

Posted: Thu Feb 27, 2020 3:17 pm
by bryanlyon

No, a TPU isn't usable, especially the edge TPU which is only made for inference and can't train.


Re: It's it possible to use edge-TPU to train a modle?

Posted: Thu Feb 27, 2020 3:19 pm
by jhsy1209021

But it semms to support tensorflow lite?


Re: It's it possible to use edge-TPU to train a modle?

Posted: Thu Feb 27, 2020 5:59 pm
by torzdf

Tensorflow Lite is inference only.

You need to be able to train models to use Faceswap.


Re: It's it possible to use edge-TPU to train a modle?

Posted: Thu Oct 07, 2021 7:15 pm
by PooOnBirbs

I was about to drop ~$50 to slap an M.2 to PCI card + install one of those Coral (Google) Edge TPU's into my rig specifically for FaceSwap... until I read this. So, outside of GPUs, are there ANY plug-and-play TPU's available that would actually be able to train in FaceSwap? Obviously, I would need to find one that supports actual TensorFlow (not TensorFlow LITE). But could you point me in the right direction? Any suggestions?


Re: It's it possible to use edge-TPU to train a modle?

Posted: Thu Oct 07, 2021 10:38 pm
by bryanlyon

No, we only support GPUs. TPUs, NPUs, the Apple M1, and any other "neural accelerators" are not supported. Most of those devices support inference ONLY and wont work for training at all. The only one we'd consider supporting is the M1 but that's hardware we can't justify purchasing and can't support without owning one.