pytorch-lightning踩坑记录

pytorch-lightning==1.5.10，python3.7，torch==1.10.0

20220421

module 'torchmetrics' has no attribute 'F1'

改成：

torchmetrics.F1Score()

You have asked for `amp_level='O2'` but it's only supported with `amp_backend='apex'

尝试添加amp_backend: apex到Trainer里，出现：

You have asked for Apex AMP but you have not installed it. Install `apex` using this guide: https://github.com/NVIDIA/apex

参考这条安装：安装apex

根据官方文档：

amp_level (Optional[str]) – The optimization level to use (O1, O2, etc…). By default it will be set to “O2” if amp_backend is set to “apex”.

expected scalar type Float but found Half

输出数据类型是torch.float32明明！

跑不下去了，先放个坑

20220430

猜测是显卡不能跑混合精度，改成precision=32，可以正常运行了

20220712

--gpus= (有几个卡就写几，要指定卡的话可以在前面加上CUDA_VISIBLE_DEVICES=0,1..指定0和1两个卡，gpus=2

pytorch-lightning踩坑记录相关推荐

树莓派下安装pytorch（踩坑记录）
目标:在树莓派4b上安装conda运行pytorch 硬件平台:树莓派4b 操作系统:Raspberry Pi OS with desktop 环境:pytorch 1. 安装系统镜像下载树莓派镜像 ...
Pytorch踩坑记录：关于用net.eval()和with no grad装饰器计算结果不一样的问题
Pytorch踩坑记录相同点 net.eval()和with toch.no_grad()的相同点:都停止反向传播不同点: 1.net.eval() 用net.eval(),此时BN层会用训练时的 ...
PyTorch踩坑记录
使用Pytorch也一年多了,记录一下自己遇到的各种问题. 各种踩坑记录 loss.backward()报错一种非常常见的错误,在网络前向传播时没问题,但是当loss.backward()时会报错 ...
TVM: Deep Learning模型的优化编译器(强烈推荐, 附踩坑记录)
本文作者是阿莱克西斯,原载于知乎,雷锋网(公众号:雷锋网)获得授权转载. (前排提醒,本文的人文内容部分稍稍带有艺术加工,请保持一定的幽默感进行阅读) 关注我最近想法的同学应该知道我最近都在把玩 TV ...
Detectron2 win10踩坑记录
Detectron2 win10踩坑记录安装CUDA clone到本地修改cocoeval.cpp第483行(不同版本行数不一样) 创建虚拟环境安装pycocotools 安装VS2019或者生 ...
Win10+yolov5 踩坑记录
Windows10下配置yolov5环境踩坑记录起因随笔基本的环境要求 yolov5各代码作用踩坑的Package windwos下torch和torchvision的安装 windows下p ...
微软NNI进行神经网络模型剪枝压缩的踩坑记录
微软NNI进行神经网络模型剪枝压缩的踩坑记录 NNI进行模型剪枝分类 NNI剪枝的流程 NNI现有剪枝方法剩下的剪枝操作比较复杂还没有研究透,不过应该大同小异,有机会继续研究更新. 最近做毕设嵌入式 ...
Nvidia Jetson TX2 详细刷机教程及踩坑记录（Jetpack3.3，python2.7，torch1.2，torchvision0.2.2）
本文总结了自带系统安装cuda等深度学习环境和使用jetpack3.3刷机的步骤,虽然自己的代码没用上,但有需要的人可以参考一下. PS:Jetpack4.5的刷机教程请移步另一篇文章: xyl-50 ...
Nvidia Jetson TX2 详细刷机教程及踩坑记录（Jetpack4.5.1，python3.6，torch1.6，torchvision0.7）
最近接触到了Nvidia Jetson TX2这块开发板,想入手测试一下自己的深度学习模型,因此和TX2的故事开始了! 前言本人刷机多次,尝试各种大小坑: 自带系统搭建环境:刚开始不想刷机,就自己手 ...
YOLO V5 CPU版本安装踩坑记录
YOLO V5 CPU版本安装踩坑记录单纯的记录下yolov5进行检测遇到了点问题的解决方法安装过程: 准备: 1,下载YOLOv5 github项目:https://github.com/ult ...

pytorch-lightning踩坑记录

pytorch-lightning踩坑记录相关推荐

最新文章

热门文章