标签: tensorflow tpu
我正在尝试在gcp jupyter笔记本(ai平台)中使用TPU api。我有自己的数据(实际上是NIH胸部X射线数据集),并且进行了很多预处理。我可以从内存中分配的numpy数组上训练GPU。我应该对此数组进行哪些转换才能成为我的tpu模型的输入。
PS:我对tpu api的看法很好,可以创建我的tpu模型。
谢谢!