NVIDIA / modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
https://developer.nvidia.com/modulus
Apache License 2.0
791 stars 165 forks source link

Fixes for EDM training #538

Closed stathius closed 1 week ago

stathius commented 1 month ago

Modulus Pull Request

Description

Fixes for #534 #535 #541 Fixed issue wandb was causing while DDP training (>=2 gpus).

Checklist

Dependencies

mnabian commented 1 month ago

/blossom-ci

mnabian commented 3 weeks ago

/blossom-ci

yairchn commented 2 weeks ago

/blossom-ci

stathius commented 1 week ago

/blossom-ci

mnabian commented 1 week ago

/blossom-ci