PyFunc operation is not supported on the current platform

I am running my code on colab+. I run CNN model correctly. But when I try tff federated avg function it starts to initialize but gives error on trainer.next(…) row. The error is: PyFunc operation is not supported on the current platform. I listed available gpu devices but no gpu devices seem. How can I solve this issue? When running eminit dataset it gives result but when I tried my custom dataset it gives error

Hi @b_dundar ,

The error you’re encountering suggests that there might be some compatibility issues between TensorFlow Federated (TFF) and the GPU environment on Colab+

TFF does not support Python functions within its computations because it needs to serialize computations for distributed execution.

You Can Refer this Article References .

Thank You .