NVIDIA / modulus-launch

Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems

Apache License 2.0

56 stars 27 forks source link

Replaced torch.norm to torch.linalg.norm to attempt to fix issue #77 #99

Closed joneseth1 closed 1 year ago

joneseth1 commented 1 year ago

Modulus Pull Request

Description

Replaced torch.norm to torch.linalg.norm to attempt to fix issue #77. Created multiple commits since I forgot to pip install 'pre-commit' so for the confusion.

mnabian commented 1 year ago

/blossom-ci

mnabian commented 1 year ago

Thanks for the PR @joneseth1! Your changes look good to me but the format test for utils.py fails. Please run black . in the modulus-launch directory and push again.

joneseth1 commented 1 year ago

Ran black . in modulus-launch and pushed. Let me know if there is anything else. Thanks so much for helping me out with my first open-source PR guys!

NickGeneva commented 1 year ago

/blossom-ci