LordMathis ef3a62f399 Remove hardcoded cuda architecture

2024-04-18 20:44:32 +02:00

..

__init__.py

Split python scripts

2024-03-22 20:03:22 +01:00

conv2d_test.py

Split python scripts

2024-03-22 20:03:22 +01:00

dense_test.py

Cleanup and refactor

2024-04-11 22:52:41 +02:00

export_model_weights.py

Remove hardcoded cuda architecture

2024-04-18 20:44:32 +02:00

gen_random_vector.py

Split python scripts

2024-03-22 20:03:22 +01:00

model_test.py

Implement pytorch weights export

2024-04-15 22:17:14 +02:00

pooling_test.py

Split python scripts

2024-03-22 20:03:22 +01:00

softmax_test.py

Add model predict test

2024-03-22 22:31:32 +01:00

utils.py

Split python scripts

2024-03-22 20:03:22 +01:00