visualizer.py - transferred 'output' attribute to cpu in get_semantic_map()

zudi-lin / pytorch_connectomics

PyTorch Connectomics: segmentation toolbox for EM connectomics

http://connectomics.readthedocs.io/

MIT License

169 stars 77 forks source link

Closed sen-28 closed 1 year ago

sen-28 commented 1 year ago

Indices should be either on cpu or on the same device as the indexed tensor