Hi team I am trying to use metaflow to train T5 with tensorflow and get a weird underlying bug

tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot convert a Tensor of dtype variant to a NumPy array.

I have tested the training codes work without metaflow. This happens in the

train

step, but after all the operation in

train

step done. So I guess this error may be related to metaflow. Does anyone have some ideas about this? Thanks! The training codes with metaflow are here, the notebook for the training without metaflow is here.