search

zhanghang1989 / PyTorch-Encoding

A CV toolkit for my papers.
https://hangzhang.org/PyTorch-Encoding/
MIT License
2.04k stars 450 forks source link

questions about parallel.py #398

Closed leileilin closed 3 years ago

leileilin commented 3 years ago

In that code, you mean that input should be already scatterd. is it achieved by using what? Thank you man.

zhanghang1989 commented 3 years ago

Please see the usage example: https://github.com/zhanghang1989/PyTorch-Encoding/blob/master/experiments/segmentation/train.py#L170-L171

leileilin commented 3 years ago

I've read this code and dataset.base.py, I didn't find the code you said. Thank u man.