Retrain on subset of chromatin profiles

[smasotti21]

I wanted to train the Sei Model only on a subset of chromatin profiles than all 21907. I was wondering what needs to be changed to make that happen. I change the seq_chromatin_profiles.txt to the ones which I am interested in (all GM12878). Further, I changed the ./data/seq_chromatin_peaks.bed.gz to have peaks only related to GM12878. I also change the./model/sei.py to have n_genomics=1. However, torch still expects to have 21907 as a size. Can you please guide me with this?

Please let me know, thank you so much!