- 博客(6)
- 收藏
- 关注
RSS订阅原创 pytorch图像分类框架搭建——利用logging记录训练日志
关于logging模块的基本使用参考Python之日志处理（logging模块），这里主要记录使用logging时踩过的坑。1. 使用pytorch的DDP进行多卡训练时如何保证只有主进程将信息发送到控制台和l日志文件中呢？logging.getLogger()方法有一个可选参数name，该参数表示将要返回的日志器的名称标识，如果不提供该参数，则其值为'root'，得到的就是RootLoggerlogger的名称是一个以'.'分割的层级结构，每个'.'后面的logger都是'.'前面的logge
2021-08-26 15:00:30
1537
原创 pytorch图像分类框架搭建——训练流程的定义xxx_train.py
训练脚本主要包含两个函数：parse_args()用来收集参数，train()定义了整个训练流程。import argparseimport timeimport tqdmimport loggingfrom pathlib import Pathfrom datetime import datetimefrom einops import rearrangeimport sysimport torchfrom torch import nnfrom torch.utils.da.
2021-08-25 15:19:39
701
原创 pytorch图像分类框架搭建——项目目录，自定义文件、包导入
最近接手的项目是关于图像分类的，参考了一些优秀的代码库，对使用pytorch框架的搭建分类项目的过程做一个记录总结，同时也记录一些常用的python包的使用。
2021-08-24 16:21:20
484
原创 图像数据集均值与标准差的计算程序（Python多线程加速）
图像深度学习任务中，一般首先要对输入图像进行归一化，本文设计了一个可以在linux服务器上使用python多线程加速对统计数据集的均值标准差的程序
2021-08-10 15:49:52
1004
原创 关于virtualenv不能创建独立环境的疑问--虚拟环境中不要sudo + pip
这两天一直想在实验室的服务器上装tensorflow，考虑到环境共享的问题，自己想利用virtualenv在自己的工作目录中创建一个独立的python环境。 问题来了，当使用pip安装的时候总会去系统的环境下去定位依赖包，这让我很郁闷，甚至对virtualenv的功能产生了严重的怀疑。后来在创建独立环境的时候加入了–no-site-packages 依然无卵用（查看man手册才知道这是默认执行的...
2018-09-11 19:59:19
580
转载 关于Visual studio 2015 未能正确加载“Microsoft.VisualStudio.Editor.Implementation.EditorPackage”包的解决方案
刚准备补一下c++，跟的学堂在线上郑莉老师的课，安装了VS2015却发现了下面的问题， 恰好找到了这位仁兄的解决方案，https://blog.csdn.net/wutianao/article/details/79427027 我用的第一种方法就解决了，希望日后还有帮助。。。。。...
2018-09-02 20:48:40
2903
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人