search

NVIDIA / VariantWorks

Deep Learning based variant calling toolkit - https://clara-parabricks.github.io/VariantWorks/
Apache License 2.0
44 stars 11 forks source link

[IO] fix multi gpu data loading #126

Closed tijyojwad closed 4 years ago

tijyojwad commented 4 years ago
  1. add torch distributed sampler for multi gpu training for both dataloaders
  2. fix consensus sample local_rank argument name needed for NeMo support