egnn
egnn copied to clipboard
Add Weights and Biases Logging Support
This PR aims to add basic Weights and Biases Metric Logging by appending to the existing training scripts with minimal changes.
The PR adds 3 extra arguments namely --use_wandb
, --wandb_project
and --wandb_entity
which can be used to specify whether to use wandb, the name of the project to be used ("egnn"
by default) and name of the entity to be used.
Request for Review @vgsatorras
Gentle ping @vgsatorras
Closing this for now, will reopen if needed 😄