Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add wandb logging support to Trainer class #166

Merged
merged 6 commits into from
Jun 13, 2024
Merged

Add wandb logging support to Trainer class #166

merged 6 commits into from
Jun 13, 2024

Commits on Jun 12, 2024

  1. add wandb logging to Trainer class

    @janosh
    janosh committed Jun 12, 2024
    Configuration menu
    Copy the full SHA
    4140672 View commit details
    Browse the repository at this point in the history

  2. add optional deps set logging = ["wandb>=0.17"] and install in CI

    @janosh
    janosh committed Jun 12, 2024
    Configuration menu
    Copy the full SHA
    265fb60 View commit details
    Browse the repository at this point in the history

  3. test wandb_path and wandb_kwargs in test_trainer

    @janosh
    janosh committed Jun 12, 2024
    Configuration menu
    Copy the full SHA
    e7ef618 View commit details
    Browse the repository at this point in the history

Commits on Jun 13, 2024

  1. only wandb.log if "wandb_path" in trainer_args

    @janosh
    janosh committed Jun 13, 2024
    Configuration menu
    Copy the full SHA
    fb2e75c View commit details
    Browse the repository at this point in the history

  2. add and test extra_run_config: dict | None = None keyword to Trainer … 

    …to specify run params like batch_size that aren't already recorded by the trainer_args dict
    @janosh
    janosh committed Jun 13, 2024
    Configuration menu
    Copy the full SHA
    bc87fd0 View commit details
    Browse the repository at this point in the history

  3. fix ruff isort.known-third-party = ["wandb"]

    @janosh
    janosh committed Jun 13, 2024
    Configuration menu
    Copy the full SHA
    508abef View commit details
    Browse the repository at this point in the history