egnn icon indicating copy to clipboard operation
egnn copied to clipboard

Add Weights and Biases Logging Support

Open SauravMaheshkar opened this issue 3 years ago • 2 comments

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.

SauravMaheshkar avatar Feb 23 '22 15:02 SauravMaheshkar

Request for Review @vgsatorras

SauravMaheshkar avatar Mar 21 '22 15:03 SauravMaheshkar

Gentle ping @vgsatorras

SauravMaheshkar avatar Jun 23 '22 17:06 SauravMaheshkar

Closing this for now, will reopen if needed 😄

SauravMaheshkar avatar Feb 19 '23 10:02 SauravMaheshkar