search

donnyyou / torchcv

TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision
https://pytorchcv.com
Apache License 2.0
2.25k stars 378 forks source link

AttributeError: 'RandomSampler' object has no attribute 'set_epoch' when det train #125

Closed qrsforever closed 5 years ago

qrsforever commented 5 years ago

I check the code in datasets/det/data_loader.py, codes missing: 

if self.configer.get('network.distributed'):                
     sampler = torch.utils.data.distributed.DistributedSampler(dataset)