instadeepai / nucleotide-transformer

🧬 Nucleotide Transformer: Building and Evaluating Robust Foundation Models for Human Genomics
https://www.biorxiv.org/content/10.1101/2023.01.11.523679v2
Other
481 stars 55 forks source link

Training process for Nucleotide Transformer Model #42

Closed ghost closed 6 months ago

ghost commented 10 months ago

Thank you for sharing Nucleotide Transformer! I am trying to analyze RNA sequences using a custom dataset that is not compatible with the available pretrained models. I would like to train the model on my own dataset for a classification task, however I am having trouble implementing a training loop that is compatible with Nucleotide Transformer. Is it possible to expand the codebase to include examples of your training procedures and implemented metrics, etc. so a new pretrained model may be built?

dallatt commented 9 months ago

Hello,

If you plan on finetuning one of the models on your custom dataset (or even training from scratch), I suggest you look at the HuggingFace notebooks we made available.

Best regards,

Hugo