Geneformer / examples /pretrain_geneformer_w_deepspeed.py

Commit History

Update pretrainer for transformers==4.28.0
b925dcc

Christina Theodoris commited on

Add data collator for cell classification and example for cell classification
088ea6e

Christina Theodoris commited on

Add Geneformer trainer and pretraining example
bcc03e8

Christina Theodoris commited on