当前位置: 首页 > news >正文

Python技巧---tqdm库的使用

文章目录

  • 一、tqdm基本知识
  • 二、在pytorch中使用tqdm


提示:以下是本篇文章正文内容,下面案例可供参考

一、tqdm基本知识

“tqdm” 是一个 Python 库,用于在命令行界面中创建进度条。
基本使用如下:

from tqdm import tqdm
import timeitems = range(10)
for item in tqdm(items, desc="Test", total=len(items)):time.sleep(1)

其文档如下:
在这里插入图片描述
只介绍传入的三个参数:iterable, desc, total
iterable:是一个可迭代对象
desc:进度条前的描述性信息
total:可迭代对象的长度
结果如下:
在这里插入图片描述
可以看到有描述性信息,进度条,已经运行了多少时间,还差多少时间,速度。在之后还可以添加后缀描述,见下面。

二、在pytorch中使用tqdm

一般都是在train函数中使用tqdm,讲dataloader做为一个可迭代对象传入tqdm

loop = tqdm((dataloader_train), desc=f"Epoch: [{epoch}/20]", total=len(dataloader_train))for img, label in loop:img = img.to(device)label = label.to(device)output = model(img)optimizer.zero_grad()loss = criterion(output,label)loss.backward()optimizer.step()train_loss += loss.item()correct += (torch.argmax(output,dim=1) == label).sum().item()loop.set_postfix(loss=loss.item() / label.shape[0])print("epoch: {i}    Train Loss: {loss}".format(i=epoch, loss=train_loss))print("epoch: {i}    Train Accuracy: {acc}".format(i=epoch, acc=correct / len(dataset_train)))

在这里插入图片描述

http://www.lryc.cn/news/167424.html

相关文章:

  • linux-线程条件变量(cond)
  • 面试算法6:排序数组中的两个数字之和
  • 【智能家居-大模型】构建未来,聆思大模型智能家居交互解决方案正式发布
  • 通讯网关软件002——利用CommGate X2HTTP-U实现HTTP访问OPC UA Server
  • 模拟经营类游戏是怎么开发的?
  • 基于JAVA+SSM+微信小程序+MySql的图书捐赠管理系统设计与实现
  • 软件设计模式系列之六——单例模式
  • verdi dump状态机的波形时直接显示状态名
  • 代码随想录算法训练营19期第53天
  • 二刷力扣--栈和队列
  • 第六章 图 十、关键路径
  • Virtualbox固定存储硬盘转换为动态存储硬盘
  • 【栈与队列面试题】有效的括号(动图演示)
  • 基于matlab实现的弹簧振动系统模型程序(动态模型)
  • 哨兵1号(Sentinel-1)SAR卫星介绍
  • [maven] scopes 管理 profile 测试覆盖率
  • css网页打印字体设置
  • JAVA高级技术入门(单元测试,反射,注解,动态代理)
  • uni-app 实现自定义按 A~Z 排序的通讯录(字母索引导航)
  • C++ PrimerPlus 复习 第一章 命令编译链接文件 make文件
  • 微信小程序——常用组件的属性介绍
  • 【深度学习】 Python 和 NumPy 系列教程(廿七):Matplotlib详解:3、多子图和布局:散点矩阵图(Scatter Matrix Plot)
  • 解决jupyter打开的默认路径问题
  • Git 学习笔记
  • 【Qt】QGroundControl入门3:源码初探
  • 腾讯mini项目-【指标监控服务重构】2023-07-31
  • Rust通用编程概念(3)
  • 学Python的漫画漫步进阶 -- 第四步
  • 【LeetCode-中等题】18. 四数之和
  • 每日一题 102二叉树的层序遍历