search

EnsemblGSOC / Ensembl-Repeat-Identification

A Deep Learning repository for predicting the location and type of repeat sequence in genome.
4 stars 3 forks source link

explore PyTorch Lightning for the training loop #26

Closed williamstark01 closed 2 years ago

williamstark01 commented 2 years ago

This is for next week when you find some time:

PyTorch Lightning might help with organizing the training loop and hyperparameters, take a look to see whether you think it may be useful. https://github.com/Lightning-AI/lightning

yangtcai commented 2 years ago

I will check it later :D