20 Commits

Author	SHA1	Message	Date
LordMathis	10e73638b6	Add non square pooling and batch norm tests	2024-05-20 22:16:00 +02:00
LordMathis	da8f3167cb	Add utils vector mean function	2024-05-14 21:41:18 +02:00
LordMathis	96536a44fd	Implement batch norm test	2024-04-28 21:46:42 +02:00
LordMathis	69111f6cb1	Add version to model bin format	2024-04-23 19:53:48 +02:00
LordMathis	d64a28bc9c	Fix model weights export	2024-04-21 00:05:56 +02:00
LordMathis	9c5d853b75	Fix bin file seek offset	2024-04-20 21:30:01 +02:00
LordMathis	0807a0f2b8	Export pretrained alexnet	2024-04-20 15:50:12 +02:00
LordMathis	ef3a62f399	Remove hardcoded cuda architecture	2024-04-18 20:44:32 +02:00
LordMathis	d8c50116e8	Implement pytorch weights export	2024-04-15 22:17:14 +02:00
LordMathis	13b455e4ba	Implement test multiple inference	2024-04-14 00:19:29 +02:00
LordMathis	457e0f669f	Implement model test	2024-04-13 13:31:29 +02:00
LordMathis	18522c2dea	Cleanup and refactor	2024-04-11 22:52:41 +02:00
LordMathis	9482d7bc43	Add model predict test	2024-03-22 22:31:32 +01:00
LordMathis	9ecd51c63a	Split python scripts	2024-03-22 20:03:22 +01:00
LordMathis	6f4cdf3792	Implement avg pool test	2024-03-20 21:57:22 +01:00
LordMathis	dfff0360d9	Implement max pooling test	2024-03-20 21:44:04 +01:00
LordMathis	48060dc38d	Rename python tool script	2024-03-17 19:11:41 +01:00
LordMathis	cbdb4e7707	Test softmax	2024-03-17 19:08:16 +01:00
LordMathis	2518138ef8	Add strided conv2d test	2024-03-11 21:51:27 +01:00
LordMathis	6bbc036f62	Generate conv2d test results with pytorch	2024-03-10 19:11:48 +01:00