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