diff --git a/cnn_tutorial.py b/cnn_tutorial.py index e164147..1b7c837 100755 --- a/cnn_tutorial.py +++ b/cnn_tutorial.py @@ -123,7 +123,7 @@ def train( # ╔─────────────────────────────────────────────────────────────────────────────╗ -# │ Mαiη scriρτ | +# │ Main script | # ╚─────────────────────────────────────────────────────────────────────────────╝ jax.config.update("jax_platform_name", "gpu") # Sets preferred device