write some tests for models and training
jeremyjordan opened this issue · comments
What's your idea?
We should improve our test coverage in this repo.
Describe the solution you'd like
Simple tests for:
- model can accept tensors of expected shape
- model can decrease the loss through a single optimization step
- feature processing code is tested