I have trained a bert model in my colab using TPU and saved the checkpoint (.ckpt.meta and .ckpt.data), how can I use it on my own computer to predict?
I have tried to use tf.train.import_meta_graph to load .ckpt.meta but it report that
tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op 'TPUReplicatedInput' used by {{node input0}}with these attrs: [T=DT_INT32, N=8]
Maybe it is caused by the trained process which uses TPU? Should I try to use model.save() to save the model as .h5 format to use it?
I have trained a bert model in my colab using TPU and saved the checkpoint (.ckpt.meta and .ckpt.data), how can I use it on my own computer to predict?
I have tried to use tf.train.import_meta_graph to load .ckpt.meta but it report that
Maybe it is caused by the trained process which uses TPU? Should I try to use model.save() to save the model as .h5 format to use it?