search

CUNY-CL / yoyodyne

Small-vocabulary sequence-to-sequence generation with optional feature conditioning
Apache License 2.0
25 stars 15 forks source link

Log model parameters to wandb #168

Closed Adamits closed 2 months ago

Adamits commented 2 months ago

Adds number of model parameters to W&B log.

Note that PTL logs this to stdout by default, but it is convenient to have a column in W&B for this, so that we do not need to retrieve and parse all stdout logs for large analysis.

Adamits commented 2 months ago

Good call, that works.

Adamits commented 2 months ago

LGTM, assuming I'm right and this works. Shall I merge?

I tested it---yes!