search

XiaLiPKU / EMANet

The code for Expectation-Maximization Attention Networks for Semantic Segmentation (ICCV'2019 Oral)
https://xialipku.github.io/publication/expectation-maximization-attention-networks-for-semantic-segmentation/
GNU General Public License v3.0
680 stars 130 forks source link

Could you please tell me where is the code to put the input data in GPU? #41

Closed caozhe1011 closed 4 years ago

caozhe1011 commented 4 years ago

I only find you put the model in gpu......thx

XiaLiPKU commented 4 years ago

pin_memory=True in https://github.com/XiaLiPKU/EMANet/blob/f09ffac3a2294881b353b43300f1295065a398a8/train.py#L69