bugfix: best_validated_encoder was loaded instead of retrained_encoder

src/maggotuba/models/modules.py

@@ -139,8 +139,9 @@ class PretrainedMaggotEncoder(MaggotEncoder):
             ptfile="best_validated_encoder.pt"):
         super().__init__(path, cfgfile, ptfile)
 
-    def save_model(self, ptfile="retrained_encoder.pt"):
-        return super().save_model(ptfile)
+    def save(self, ptfile="retrained_encoder.pt"):
+        self.ptfile = ptfile
+        return super().save()
 
 class MaggotEncoders(nn.Module):
     def __init__(self, paths, cls=MaggotEncoder, **kwargs):