Tensor extension for snoop
Project description
snoop-tensor
Numpy/Pytorch Tensor extension for snoop. It's modified from TorchSnooper
Usage
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
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
snoop-tensor-0.2.tar.gz
(4.7 kB
view hashes)