Model-Swap-Face_v2 icon indicating copy to clipboard operation
Model-Swap-Face_v2 copied to clipboard

Published 20 hours ago • verified publisher icon a312863063

Can we train the model on TPU?

Open JANVI2411 opened this issue 3 years ago • 0 comments

I want to train the model for African Ethnicity on Colab TPU. But when I do it, "from op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d" is wrong (No CUDA Home ./local/..) in "model.py".

I guess it may be an issue in folder "op" since it defines c++ extension.

would you mind taking some advice for me ?

JANVI2411 avatar Feb 24 '22 12:02 JANVI2411