issues
search
jasonkyuyim
/
se3_diffusion
Implementation for SE(3) diffusion model with application to protein backbone generation
https://arxiv.org/abs/2302.02277
MIT License
305
stars
50
forks
source link
Fix __len__() of the `TrainSampler` class
#26
Closed
amorehead
closed
1 year ago
amorehead
commented
1 year ago
Keeps the
len
value of the
TrainSampler
class up-to-date w/ or w/o clustering. Otherwise, the
TrainSampler
will cause an
OutOfIndexError
downstream for e.g., Lightning's
_DatasetSamplerWrapper
class (
https://github.com/Lightning-AI/lightning/blob/23776b0dedf2bccdc97b21e47029db786b8bd8b1/src/lightning/fabric/utilities/distributed.py#L284
).
lenvalue of theTrainSamplerclass up-to-date w/ or w/o clustering. Otherwise, theTrainSamplerwill cause anOutOfIndexErrordownstream for e.g., Lightning's_DatasetSamplerWrapperclass (https://github.com/Lightning-AI/lightning/blob/23776b0dedf2bccdc97b21e47029db786b8bd8b1/src/lightning/fabric/utilities/distributed.py#L284).