Implement Tensorflow variant utilizing GPU

2022-07-27 12:01:23 +02:00
parent aebd885ab7
commit fa37c40f0b
2 changed files with 133 additions and 2 deletions
@@ -34,9 +34,9 @@ def feed_forward(x: np.ndarray, *layers: tuple[np.ndarray, np.ndarray]) -> np.nd
    return x


-def main(n):
+def main(n: int) -> None:
    t = timeit(
-        'feed_forward(inp, *layers)',
+        stmt='feed_forward(inp, *layers)',
        setup='from __main__ import feed_forward, load_test_data;' + 
              'inp, layers = load_test_data()', 
        number=n