Skip to content

Commit

Permalink
update example notebook
Browse files Browse the repository at this point in the history
  • Loading branch information
akensert committed Jun 21, 2024
1 parent 784e169 commit bb6aaed
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion notebooks/examples-masking.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@
" featurizers.Conjugated()\n",
"]\n",
"\n",
"# Currently, collate_fn performs the masking. Should it be performed in PeptideGraphEncoder instead?\n",
"# Currently, collate_fn performs the masking.\n",
"# ***IMPORTANT***: When fine-tuning a pretrained model, the PeptideGraphEncoder for the fine-tuning task \n",
"# also needs supports_masking=True (as dimensions need to match).\n",
"peptide_graph_encoder = encoders.PeptideGraphEncoder(\n",
Expand Down

0 comments on commit bb6aaed

Please sign in to comment.