textgenrnn icon indicating copy to clipboard operation
textgenrnn copied to clipboard
verified publisher icon minimaxir

TPU Support?

Open aidanmclaughlin opened this issue 6 years ago • 2 comments

Is there a way to use a tensor processing unit for acceleration? If not, is this feature going to be added in the future?

aidanmclaughlin avatar Jul 27 '19 21:07 aidanmclaughlin

it only uses CPU

breadbrowser avatar Jul 06 '22 00:07 breadbrowser

no gpu or tpu needed also it would train at the same speed.

breadbrowser avatar Jul 06 '22 00:07 breadbrowser