FAIR-Chem / fairchem

FAIR Chemistry's library of machine learning methods for chemistry
https://opencatalystproject.org/
Other
774 stars 243 forks source link

Hide wandb.watch behind flag #747

Closed rayg1234 closed 3 months ago

rayg1234 commented 3 months ago

wanbd.watch logs too much data and makes the UI slow, turn it off by default.

Example usage (config in yaml):

logger:
  name: wandb
  entity: fairchem
  project: fm_testing
  watch: 1000

the number specifies the frequency (in # of steps) to log gradients etc..

Run without watch: https://fairwandb.org/fairchem/fm_testing/runs/2024-07-02-16-12-48-local_test_no_watch Run with watch: https://fairwandb.org/fairchem/fm_testing/runs/2024-07-02-16-10-40-local_test_watch

codecov[bot] commented 3 months ago

Codecov Report

Attention: Patch coverage is 66.66667% with 2 lines in your changes missing coverage. Please review.

Files Coverage Δ
src/fairchem/core/common/logger.py 72.22% <75.00%> (-3.97%) :arrow_down:
src/fairchem/core/trainers/base_trainer.py 86.26% <50.00%> (-0.19%) :arrow_down:

... and 1 file with indirect coverage changes