diff options
Diffstat (limited to 'transquest/algo/sentence_level/siamesetransquest/models/Pooling.py')
-rw-r--r-- | transquest/algo/sentence_level/siamesetransquest/models/Pooling.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/transquest/algo/sentence_level/siamesetransquest/models/Pooling.py b/transquest/algo/sentence_level/siamesetransquest/models/Pooling.py index 956048b..efafce4 100644 --- a/transquest/algo/sentence_level/siamesetransquest/models/Pooling.py +++ b/transquest/algo/sentence_level/siamesetransquest/models/Pooling.py @@ -80,7 +80,7 @@ class Pooling(nn.Module): return {key: self.__dict__[key] for key in self.config_keys} def save(self, output_path): - with open(os.path.join(output_path, 'config.json'), 'w') as fOut: + with open(os.path.join(output_path, 'pooling_config.json'), 'w') as fOut: json.dump(self.get_config_dict(), fOut, indent=2) @staticmethod |