Great work putting together this project!
Two minor indexing errors in the TRRMSADataset class:
l256 in data.py:
sliced_msa = msa[:, slice_start: slice_start + self.max_seq_len] --> sliced_msa = msa[:, slice_start: slice_start + seq_len]
l274 in data.py:
output = sliced_msa[:64] --> output = sliced_msa[:self.n_sequences]
Great work putting together this project! Two minor indexing errors in the TRRMSADataset class: l256 in data.py:
sliced_msa = msa[:, slice_start: slice_start + self.max_seq_len]
-->sliced_msa = msa[:, slice_start: slice_start + seq_len]
l274 in data.py:
output = sliced_msa[:64]
-->output = sliced_msa[:self.n_sequences]