![](https://img-blog.csdnimg.cn/20190918140145169.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
ML&DL
ee
aisuperdoger
这个作者很懒,什么都没留下…
展开
-
requires_grad,grad_fn,grad的含义及使用
因为pytorch中规定只要某一个输入需要相关梯度值,则输出也需要保存相关梯度信息,这样就保证了这个输入的梯度回传,即我们只需要设置x的requires_grad=True,就可以保证y、z、out的requires_grad=True。还可以看出,要求out对x的偏导,需要知道out对z的偏导、z对y的偏导、y对x的偏导。我觉得requires_grad设为True的作用就是保存这些偏导信息(也称为梯度信息),比如y的requires_grad设为True,则会保存“y对x的偏导”这个信息。......原创 2022-07-18 15:50:10 · 2129 阅读 · 0 评论 -
python提取COCO数据集中特定的类
import osfrom shutil import copyfile,movesrc_path = './labels/train2017/' # 标签img_path = './images/train2017/' # 图像dst_label_path = './traffic_train_coco/labels/'dst_img_path = './traffic_train_coco/images/'cls_id = ['79'] # 牙刷labels = os.listd转载 2022-05-17 21:11:06 · 313 阅读 · 0 评论 -
1×1的卷积核
卷积核都有增加非线性。1×1的卷积核只在通道上进行处理,其他的卷积核是在通道和面上进行处理。原创 2022-01-13 21:26:24 · 1399 阅读 · 0 评论 -
白嫖GPU
这里首选免费深度学习平台——colab。Colab的使用:https://blog.csdn.net/qq_36890370/article/details/106160805Colab的使用的补充:https://zhuanlan.zhihu.com/p/54389036原创 2021-09-25 10:03:23 · 88 阅读 · 0 评论 -
ubuntu源码安装pytorch——源码编译(包含驱动,cuda,cudnn安装)
遇到的错误在源码安装pytorch时,我的cuda的版本时11.4,所以按照官方的安装指南是需要magma-cuda114,而此时magma-cuda114还没发布。所以我的方法是卸载驱动、CUDA和CUDNN(具体自己百度)。然后选择对应cuda版本为11.3的显卡驱动,然后安装cuda11.3和相应的CUDNN。说明:显卡的驱动可以用较低版本的。......原创 2021-10-07 21:10:39 · 1768 阅读 · 0 评论 -
yolov5使用TensorRT进行加速
https://www.mdnice.com/writing/328959e2439045849a06933c6380e148原创 2021-10-03 11:16:28 · 431 阅读 · 0 评论 -
tensorboard的使用
TensorBoard 是 TensorFlow提供的一组可视化工具(a suite of visualization tools),可以帮助开发者方便的理解、调试、优化TensorFlow 程序from torch.utils.tensorboard import SummaryWriterimport numpy as npfrom PIL import Imagewriter = SummaryWriter("logs")image_path = "img.png"img_PIL = I原创 2021-09-25 10:34:42 · 59 阅读 · 0 评论