Common Utils for PyTorch
Project description
torch-utils

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.
Filename, size | File type | Python version | Upload date | Hashes |
---|---|---|---|---|
Filename, size torch-utils-0.1.2.tar.gz (4.9 kB) | File type Source | Python version None | Upload date | Hashes View |