|
|
757584544c
|
Migrate conv2d layer
|
2024-09-10 21:19:50 +02:00 |
|
|
|
74f49d6a00
|
Migrate output layer
|
2024-09-10 19:20:00 +02:00 |
|
|
|
f7b525e494
|
Migrate input layer
|
2024-09-10 19:11:21 +02:00 |
|
|
|
fe7c16ac36
|
Migrate concat layer
|
2024-09-09 22:16:22 +02:00 |
|
|
|
a0665fb05c
|
Migrate max pooling
|
2024-09-09 21:48:36 +02:00 |
|
|
|
76e5225001
|
Migrate avg pooling
|
2024-09-09 21:36:13 +02:00 |
|
|
|
75475790ac
|
Migrate Dense layer
|
2024-09-08 13:36:53 +02:00 |
|
|
|
591507ed21
|
Split tests to cpu and cuda
|
2024-09-07 21:24:34 +02:00 |
|
|
|
9704d0d53e
|
Add running mean and running var to batchnorm
|
2024-08-25 19:05:10 +02:00 |
|
|
|
7f745027cd
|
Implement BasicConv2d test
|
2024-06-02 14:40:46 +02:00 |
|
|
|
8168f02f58
|
Add adaptive avg pooling
|
2024-05-30 17:17:31 +02:00 |
|
|
|
df47a31f36
|
Rename dim2d to shape2d
|
2024-05-27 21:14:51 +02:00 |
|
|
|
c09e978308
|
Remove dense init test
|
2024-05-26 19:04:04 +02:00 |
|
|
|
94a16b4352
|
Add padding to max pooling
|
2024-05-26 19:03:10 +02:00 |
|
|
|
4a67b708f0
|
Add padding to avg pooling
|
2024-05-26 18:54:12 +02:00 |
|
|
|
78a0fd0baf
|
Add non square test to conv2d
|
2024-05-23 21:50:39 +02:00 |
|
|
|
a300ab6dd5
|
Refactor conv2d test
|
2024-05-23 20:31:54 +02:00 |
|
|
|
10e73638b6
|
Add non square pooling and batch norm tests
|
2024-05-20 22:16:00 +02:00 |
|
|
|
6dca8ccd3c
|
Unify 2d layer naming
|
2024-05-20 16:23:58 +02:00 |
|
|
|
74098b24e3
|
Add support for non square matrices
|
2024-05-20 15:20:43 +02:00 |
|
|
|
6f8b5f4081
|
Rename batchnorm
|
2024-05-20 13:05:48 +02:00 |
|
|
|
4a1c4a5f91
|
Add epsilon param to batch norm
|
2024-05-19 15:13:22 +02:00 |
|
|
|
33a3c6afcb
|
Fix batch norm layer
|
2024-05-14 21:42:01 +02:00 |
|
|
|
da8f3167cb
|
Add utils vector mean function
|
2024-05-14 21:41:18 +02:00 |
|
|
|
96536a44fd
|
Implement batch norm test
|
2024-04-28 21:46:42 +02:00 |
|
|
|
69111f6cb1
|
Add version to model bin format
|
2024-04-23 19:53:48 +02:00 |
|
|
|
f17debc244
|
Implement getOutputSize and getInputSize for seq layers
|
2024-04-22 20:31:58 +02:00 |
|
|
|
fbf6c44bdd
|
Remove extra cudaFree
|
2024-04-22 19:05:02 +02:00 |
|
|
|
26cea9b12c
|
Fix more cuda memcheck errors
|
2024-04-21 22:56:55 +02:00 |
|
|
|
58af95eb25
|
Remove cudaDeviceReset calls from tests
|
2024-04-21 22:47:12 +02:00 |
|
|
|
bdbb3f978e
|
Fix matmul and max reduce memcheck errors
|
2024-04-21 22:11:02 +02:00 |
|
|
|
9a6152469a
|
Update activation test
|
2024-04-21 14:00:43 +02:00 |
|
|
|
d64a28bc9c
|
Fix model weights export
|
2024-04-21 00:05:56 +02:00 |
|
|
|
ecf7416f8e
|
Rework padding size setting
|
2024-04-20 16:31:28 +02:00 |
|
|
|
432adf57bd
|
Test model weights loading
|
2024-04-16 21:07:06 +02:00 |
|
|
|
f4ae45f867
|
Start implementing weights import
|
2024-04-15 22:17:48 +02:00 |
|
|
|
d8c50116e8
|
Implement pytorch weights export
|
2024-04-15 22:17:14 +02:00 |
|
|
|
13b455e4ba
|
Implement test multiple inference
|
2024-04-14 00:19:29 +02:00 |
|
|
|
53c976733b
|
Refactor model test
|
2024-04-14 00:05:32 +02:00 |
|
|
|
457e0f669f
|
Implement model test
|
2024-04-13 13:31:29 +02:00 |
|
|
|
18522c2dea
|
Cleanup and refactor
|
2024-04-11 22:52:41 +02:00 |
|
|
|
4b9d123e94
|
Implement device vector utils
|
2024-04-11 22:22:33 +02:00 |
|
|
|
710a33bdde
|
Move softmax partial kernels to matmul
|
2024-04-11 22:01:47 +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 |
|
|
|
e419a93408
|
Fix softmax sum kernel
|
2024-04-08 22:09:18 +02:00 |
|
|
|
7bc329a043
|
Add more softmax tests
|
2024-03-22 22:32:08 +01:00 |
|
|
|
9482d7bc43
|
Add model predict test
|
2024-03-22 22:31:32 +01:00 |
|
|
|
90fb104dae
|
Implement output layer
|
2024-03-21 23:07:46 +01:00 |
|
|
|
6f4cdf3792
|
Implement avg pool test
|
2024-03-20 21:57:22 +01:00 |
|
