search

srush / annotated-s4

Implementation of https://srush.github.io/annotated-s4
https://srush.github.io/annotated-s4
MIT License
468 stars 60 forks source link

Add Weights and Biases Logging #42

Closed SauravMaheshkar closed 2 years ago

SauravMaheshkar commented 2 years ago

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 ("s4" by default) and name of the entity to be used.

srush commented 2 years ago

Nice. I think this is good to add. It's convenient that it doesn't need to touch the main blog post.