Set unit and layer default values consistently

[timokau]

Description

Always use 2 layers, 32 units by default.

Fixes #175

Motivation and Context

175

How Has This Been Tested?

CI

Does this close/impact existing issues?

Fixes #175.

Types of changes

• [x Bug fix (non-breaking change which fixes an issue)
• [ ] New feature (non-breaking change which adds functionality)
• [x] Breaking change (fix or feature that would cause existing functionality to change) (kind of)

Checklist:

• [x] My code follows the code style of this project.
• [ ] My change requires a change to the documentation.
• [ ] I have updated the documentation accordingly.
• [ ] I have added tests to cover my changes.

[kiudee]

Looks like some of the tests need to be adjusted due to the new default parameters.

[timokau]

I relied on the CI to run the tests. Should've seen the failure coming here though. Fixed now, please have another look.