Common Utils for PyTorch
Project description
torch-utils
![PyPI version](https://pypi-camo.freetls.fastly.net/e161563e7faf20b8e583e7e93092aba3f5ac218c/68747470733a2f2f62616467652e667572792e696f2f70792f746f7263682d7574696c732e737667)
Common Utils for PyTorch.
Installation
Need Python 3.6+.
pip install torch-utils
Usage
- Accuracy
import torch_utils
# ...
top_1, top_5 = torch_utils.accuracy(output=..., target=..., top_k=(1, 5))
- Meter
import torch_utils
loss_meter = torch_utils.AverageMeter(name='Meter', length=10)
loss_meter.update(val=...)
print(loss_meter.avg, loss_meter.val)
print(loss_meter)
#> Test 0.00 (0.00)
progress_meter = torch_utils.ProgressMeter(total_steps=100, total_epochs=10)
progress_meter.update(step=10)
assert progress_meter.step == 10
assert progress_meter.ratio == 0.1
assert progress_meter.epoch == 1
print(progress_meter)
#> Step 10/100=10.0% (1/10)
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
torch-utils-0.1.2.tar.gz
(4.9 kB
view hashes)