An open API service indexing awesome lists of open source software.

https://github.com/cmpute/snoop-tensor

Numpy/Pytorch Tensor extension for snoop
https://github.com/cmpute/snoop-tensor

Last synced: about 2 months ago
JSON representation

Numpy/Pytorch Tensor extension for snoop

Awesome Lists containing this project

README

          

# snoop-tensor
Numpy/Pytorch Tensor extension for snoop. It's modified from [TorchSnooper](https://github.com/zasdfgbnm/TorchSnooper)

# Usage

```python
import snoop, snoop_tensor
snoop.install()
snoop_tensor.register() # need to be called after install()
```

## Features

- Display tensors in a beautiful way
- Profiling GPU memory usage and time usage