diff --git a/CHANGES.rst b/CHANGES.rst index ef7e00f..f4486c7 100644 --- 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 index cf7f2f8..b0f77a3 100644 --- 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 index 7aaba57..ddf1271 100644 --- 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',