Commit Graph

3 Commits

Author SHA1 Message Date
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