openfl icon indicating copy to clipboard operation
openfl copied to clipboard

Add support for PyTorch 2.0+ in OpenFL

Open psfoley opened this issue 2 years ago • 0 comments

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.

psfoley avatar Apr 03 '23 18:04 psfoley