main.py

Jose Ignacio Robledo authored 4 months ago and Jan Ebert committed 4 months ago

Since the default PyTorch module is now 2.5, we can finally integrate
the previously commented changed, enabling newer APIs and fixing
deprecation warnings.

Closes !1.