Skip to content

Commit

Permalink
fixed lint issue
Browse files Browse the repository at this point in the history
  • Loading branch information
Isha Gokhale committed Oct 17, 2023
1 parent b7c2ff8 commit 8136a4b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion casanovo/denovo/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down

0 comments on commit 8136a4b

Please sign in to comment.