Solved problem with default top module in ShiHaotian. Closes #3

ari-dasci · Nov 8, 2023 · 201836a · 201836a
1 parent 3bd3867
commit 201836a
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -133,3 +133,15 @@ latest versions.
 - Deprecated:
 - Removed:
 - Fixed:
+
+1.0.7.5 (8 Nov 2023)
+======================
+**Description**
+
+Fixed error with ShiHaotian model in Pytorch and top module. Solved and reported in issue #3.
+
+- Added:
+- Changed:
+- Deprecated:
+- Removed:
+- Fixed: ShiHaotian model in Pytorch had a default top module behavior
diff --git a/TSFEDL/models_pytorch.py b/TSFEDL/models_pytorch.py
@@ -2227,8 +2227,6 @@ def __init__(self,
                  **kwargs
                  ):
         super(ShiHaotian, self).__init__(in_features, top_module, loss, metrics, optimizer, **kwargs)
-        if self.classifier is None:
-            self.classifier = ShiHaotian_Classifier(7904, 5, in_features)
 
         self.convolutions1 = nn.Sequential(
             nn.Conv1d(in_channels=in_features, out_channels=32, kernel_size=13, stride=2, bias=True, padding="valid"),

diff --git a/setup.py b/setup.py
@@ -21,7 +21,7 @@
     name='TSFEDL',
 
     # Version
-    version='1.0.7.4',
+    version='1.0.7.5',
 
     # Description
     description='Time Series Spatio-Temporal Feature Extraction using Deep Learning',