diff --git a/casanovo/denovo/model.py b/casanovo/denovo/model.py index ca737ccd..c26529de 100644 --- a/casanovo/denovo/model.py +++ b/casanovo/denovo/model.py @@ -131,7 +131,7 @@ def __init__( dropout=dropout, residues=residues, max_charge=max_charge, - ) + ) self.softmax = torch.nn.Softmax(2) self.celoss = torch.nn.CrossEntropyLoss(ignore_index=0) # Optimizer settings.