Skip to content
Snippets Groups Projects
Commit 7fc8d0c0 authored by François  LAURENT's avatar François LAURENT
Browse files

bugfix: best_validated_encoder was loaded instead of retrained_encoder

parent 0bffd4ec
No related branches found
No related tags found
No related merge requests found
......@@ -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):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment