Hi! I have a question regarding flyte support for ...
# ask-the-community
y
Hi! I have a question regarding flyte support for tensorflow data. I have generated a
tf.data.Dataset
in one of the tasks and return it as input for next task, but got data type error (when pickle serialize this data)`tensorflow.python.framework.errors_impl.InvalidArgumentError'>: Cannot convert a Tensor of dtype variant to a NumPy array` , I'm kind of new to both tensorflow and flyte, so wondering is there any guidance on how to resolve this issue?
the tf.data.Dataset look sth like this:
Copy code
<PrefetchDataset element_spec=({'feature1': TensorSpec(shape=(32,), dtype=tf.float32, name=None), 'feature2': TensorSpec(shape=(32,), dtype=tf.float32, name=None)})>
k
could you share code snippet. that will be helpful
130 Views