Commit Graph

5 Commits

Author SHA1 Message Date
457e0f669f Implement model test 2024-04-13 13:31:29 +02:00
18522c2dea Cleanup and refactor 2024-04-11 22:52:41 +02:00
bf7c961b9e Add cudaDeviceReset at the end of each test 2024-04-11 19:55:02 +02:00
b49dddf34a Improve softmax numerical stability 2024-04-08 23:25:46 +02:00
9482d7bc43 Add model predict test 2024-03-22 22:31:32 +01:00