Skip to content
Snippets Groups Projects

Set of commits to be tagged v0.19

Merged François LAURENT requested to merge dev into main
5 files
+ 38
8
Compare changes
  • Side-by-side
  • Inline
Files
5
@@ -184,7 +184,7 @@ class MaggotEncoder(MaggotModule):
@@ -184,7 +184,7 @@ class MaggotEncoder(MaggotModule):
_reason = e
_reason = e
config['load_state'] = False # for `was_pretrained` to properly work
config['load_state'] = False # for `was_pretrained` to properly work
else:
else:
logging.debug(f"loading neural network state: {path}")
logging.debug(f"neural network state loaded: {path}")
else:
else:
_reason = '"load_state" is set to false'
_reason = '"load_state" is set to false'
# if state file not found or config option "load_state" is False,
# if state file not found or config option "load_state" is False,
@@ -336,8 +336,8 @@ class DeepLinear(nn.Module):
@@ -336,8 +336,8 @@ class DeepLinear(nn.Module):
return self.layers(x)
return self.layers(x)
def load(self, path):
def load(self, path):
logging.debug(f"loading neural network state: {path}")
self.load_state_dict(torch.load(path))
self.load_state_dict(torch.load(path))
 
logging.debug(f"neural network state loaded: {path}")
def save(self, path):
def save(self, path):
logging.debug(f"saving neural network state to file: {path}")
logging.debug(f"saving neural network state to file: {path}")
Loading