From f9daded96d9925d064f885382c2a23fb02d4042c Mon Sep 17 00:00:00 2001 From: BIGAUD Nathan <nathan.bigaud@inria.fr> Date: Mon, 13 Mar 2023 12:09:44 +0100 Subject: [PATCH] Correcting tf file typos --- test/model/test_tflow.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/model/test_tflow.py b/test/model/test_tflow.py index 2e89fb4c..a6da92e7 100644 --- a/test/model/test_tflow.py +++ b/test/model/test_tflow.py @@ -17,8 +17,8 @@ """Unit tests for TensorflowModel.""" -import warnings import sys +import warnings from typing import Any, List, Literal import numpy as np @@ -38,7 +38,7 @@ from declearn.utils import set_device_policy # dirty trick to import from `model_testing.py`; # pylint: disable=wrong-import-order, wrong-import-position sys.path.append(".") -from model_testing import ModelTestSuite, ModelTestCase +from model_testing import ModelTestCase, ModelTestSuite class TensorflowTestCase(ModelTestCase): @@ -51,7 +51,7 @@ class TensorflowTestCase(ModelTestCase): 16-neurons fully-connected layer with ReLU 1 output neuron with sigmoid activation * "MLP-tune": - - same as NLP, but freeze the first layer of the stack + - same as MLP, but freeze the first layer of the stack * "RNN": - input: 128-tokens-sequence in a 100-tokens-vocabulary - stack: 32-dimensional embedding matrix -- GitLab