ProtFlash

ProtFlash copied to clipboard

batchConverter uses up a lot of RAM

10

Is there a way to run batchConverter in a way to not use up a lot of RAM? When I'm trying to run it, it uses up all RAM and...

joelmeili

**Describe the bug** `ProtFlash` when wrapped with `torch.nn.DataParallel` crashes via an einops error. **To Reproduce** ```python import torch from ProtFlash.pretrain import load_prot_flash_base from ProtFlash.utils import batchConverter # Proteins we want...

ayaanhossain