Giters

securefederatedai / openfl

An open framework for Federated Learning.

Home Page:https://openfl.readthedocs.io/en/latest/index.html

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add support for PyTorch 2.0+ in OpenFL

psfoley opened this issue · comments

commented

This task involves adding support for PyTorch 2.0+ in the existing PyTorchTaskRunner. Certain models during the forward pass may fail due to major changes in the torch.compile(...) method.