Add a decoder file

[YouyaZhang]

No description provided.

[andre15silva]

LGMT!

Also tip: it's a good practice to have some logging setup such that you can inspect what's happening in your training runs. Could you create an account on W&B so that I can add you to our team there (https://wandb.ai/assert-kth). Then you can use their library (https://pypi.org/project/wandb/) to log and plot important metrics (e.g., loss vs. steps).

[YouyaZhang]

LGMT!

Also tip: it's a good practice to have some logging setup such that you can inspect what's happening in your training runs. Could you create an account on W&B so that I can add you to our team there (https://wandb.ai/assert-kth). Then you can use their library (https://pypi.org/project/wandb/) to log and plot important metrics (e.g., loss vs. steps).

Thanks for the suggestion! I’ve registered a W&B account. I’ll send you my account details via email.

[andre15silva]

Great! I merged the PR into main :)