ChEB-AI / python-chebai

GNU Affero General Public License v3.0
11 stars 4 forks source link

improved wandb hyperparameter tracking + tagging #30

Closed sfluegel05 closed 2 months ago

sfluegel05 commented 3 months ago

The wandb logger will now log more extensively, including

A wandb_logger.yml that uses these features might look like this:

class_path: chebai.loggers.custom.CustomLogger # Extension of Wandb logger
init_args:
  save_dir: logs
  entity: 'chebai'
  project: 'simons_personal_project'
  log_model: 'all'
  verbose_hyperparameters: True
  tags:
    - "run for demonstrating tags"
    - "different tags can be combined"