search

marian-nmt / marian-regression-tests

Regression tests for the marian-dev repository
Other
4 stars 9 forks source link

Updating training/restoring/optimizer regression tests expected keys #88

Closed alexandremuzio closed 1 year ago

alexandremuzio commented 1 year ago

Adding a new line to the adagrad.keys.expected and adam.keys.expected files with the number of parameters. The marian-dev model_info.py script is being updated to also print the model's number of parameters.

E.g.

Total number of parameters: 32603522
alexandremuzio commented 1 year ago

Adding @snukky (not sure if you are automatically added as a reviewer)