Commit Graph

6 Commits

Author SHA1 Message Date
TJU_Lu
09e832c829 fix issue: depth occasionally has negative numbers 2024-12-23 11:05:45 +08:00
TJU_Lu
43cf7bea6d Replace Tanh with ReLU of scores and simplify matrix operations 2024-12-22 15:43:04 +08:00
Lu Junjie
84bfe2e761 Fix the sorting issue when datasets path larger than 10 2024-11-03 22:52:10 +08:00
Lu Junjie
8142087649 fix a warning issue and update readme 2024-11-02 12:25:00 +08:00
Lu Junjie
6a7aba86dc Organize unused code 2024-10-20 23:40:36 +08:00
Lu Junjie
5738088bae Initial Commit (tested training, testing, and TRT conversion) 2024-10-20 17:01:07 +08:00