search

mlcommons / GaNDLF

A generalizable application framework for segmentation, regression, and classification using PyTorch
https://gandlf.org
Apache License 2.0
150 stars 78 forks source link

[FEATURE] Upgrade PyTorch to 2.2.1 #856

Closed sarthakpati closed 2 months ago

sarthakpati commented 2 months ago

Is your feature request related to a problem? Please describe.

PyTorch 2.2 has been released since 30th Jan, and it would be good to update the dependency to reflect this. Full release notes are here.

Describe the solution you'd like

Update the requirements.

Describe alternatives you've considered

N.A.

Additional context

N.A.

scap3yvt commented 2 months ago

Please assign this to me.

sarthakpati commented 2 months ago

Should be done with #857