rnajena / bertax_training

Training scripts for BERTax
8 stars 4 forks source link

added multi_tax support #4

Closed flomock closed 4 years ago

flomock commented 4 years ago

note this automatically casts the classes from the dataset. For know the different taxa predicted are: Superkingdom, Kingdom, Family. Kingdom leads to numerous 'unknown' sequences, maybe phylum would be better

furthermore minor stuff added multi gpu support (only pretraining for now), mixed precision, and minor stuff