Made device tensor accessable as host Numpy array in utils/monitor.py

zudi-lin / pytorch_connectomics

PyTorch Connectomics: segmentation toolbox for EM connectomics

http://connectomics.readthedocs.io/

MIT License

171 stars 77 forks source link

Made device tensor accessable as host Numpy array in utils/monitor.py #85

Closed ajbrookhouse closed 2 years ago

ajbrookhouse commented 2 years ago

There was a line in utils/monitor.py that was trying to access a device tensor. I made a simple change to a version of the tensor that is usable on the host as a numpy array.