# torch-utils [![Build Status](https://travis-ci.com/FebruaryBreeze/torch-utils.svg?branch=master)](https://travis-ci.com/FebruaryBreeze/torch-utils) [![codecov](https://codecov.io/gh/FebruaryBreeze/torch-utils/branch/master/graph/badge.svg)](https://codecov.io/gh/FebruaryBreeze/torch-utils) [![PyPI version](https://badge.fury.io/py/torch-utils.svg)](https://pypi.org/project/torch-utils/)
Common Utils for PyTorch.
## Installation
Need Python 3.6+.
```bash
pip install torch-utils
```
## Usage
1. Accuracy
```python
import torch_utils
# ...
top_1, top_5 = torch_utils.accuracy(output=..., target=..., top_k=(1, 5))
```
2. Meter
```python
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)
```
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
共13个文件
py:5个
txt:4个
pkg-info:2个
资源分类:Python库 所属语言:Python 资源全名:torch-utils-0.0.9.tar.gz 资源来源:官方 安装方法:https://lanzao.blog.csdn.net/article/details/101784059
资源详情
资源评论
资源推荐
收起资源包目录
torch-utils-0.0.9.tar.gz (13个子文件)
torch-utils-0.0.9
PKG-INFO 278B
torch_utils.egg-info
PKG-INFO 278B
requires.txt 6B
SOURCES.txt 287B
top_level.txt 12B
dependency_links.txt 1B
setup.cfg 38B
setup.py 491B
torch_utils
module_tag.py 2KB
__init__.py 186B
meter.py 3KB
accuracy.py 769B
README.md 1KB
共 13 条
- 1
挣扎的蓝藻
- 粉丝: 13w+
- 资源: 15万+
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功
评论0