how to train on TPU? should I migrate keras to tf.keras?

fizyr / keras-maskrcnn

Keras implementation of MaskRCNN object detection.

Apache License 2.0

406 stars 131 forks source link

how to train on TPU? should I migrate keras to tf.keras? #110

Open Tyushang opened 4 years ago

Tyushang commented 4 years ago

I want to train this model on TPU, with TensorFlow 2.1.0. I try to do this, but it seems there's compatability problems between keras and tensorflow-backend. should I migrate keras to tf.keras? Would you make a branch to achive this?