摘要: BN层在激活层之前,让模型量化损失最小。 阅读全文
posted @ 2024-11-12 11:36 张幼安 阅读(5) 评论(0) 推荐(0) 编辑
摘要: 1. tf.keras.layers.Reshape会变成一个layer, 而tf.reshape只是一个计算;用tf.reshape训练假设得到100层的模型,tf.keras.reshape训练五层会有105层模型; 2. tf.keras.layers.Reshape会消耗更多的算力和内存; 阅读全文
posted @ 2024-11-12 09:59 张幼安 阅读(13) 评论(0) 推荐(0) 编辑
摘要: 先简单记录版本,以后再详细; haotian-liu/LLaVA at v1.2.1 (github.com) 当前调通版本: cuda11.7; cudnn LLava-1.2.1; python3.10; torch2.0.1 flash-attn; flash_attn-2.0.7+cu117 阅读全文
posted @ 2024-10-11 11:33 张幼安 阅读(97) 评论(0) 推荐(0) 编辑
摘要: 做模型部署边缘设备的时候,我们经常会遇到特定格式的要求。但常见的onnx2tf很多时候都不能满足我们的要求。因此,记录一下我的操作过程。 1. 环境:(linux18.04) # Name Version Build Channel _libgcc_mutex 0.1 main defaults _ 阅读全文
posted @ 2024-06-05 14:11 张幼安 阅读(482) 评论(0) 推荐(0) 编辑
摘要: 背景如下:我要删除Round节点, 同时看了一下,Dequantize和Quantize也是没有必要的。所以最好一起删除。 原始项目地址:PINTO0309/hand-gesture-recognition-using-onnx: This is a hand gesture recognition 阅读全文
posted @ 2024-05-30 20:43 张幼安 阅读(17) 评论(0) 推荐(0) 编辑
摘要: 参考:使用 SavedModel 格式 | TensorFlow Core (google.cn) (持续更新) SavedModel是一个包含序列化签名和运行这些签名所需的状态的目录,其中包含变量值和词汇表。 $ ls {mobilenet_save_path} assets fingerprin 阅读全文
posted @ 2024-05-30 19:33 张幼安 阅读(43) 评论(0) 推荐(0) 编辑
摘要: 踩坑: 将slim和models路径加入虚拟环境中; 将slim和models路径加入系统路径中;(加入系统路径的时候,这个pwd) qit (持续更新) 具体安装流程: (很多问题)2.0版本 ;(继续更新) conda create -n tf_obj_det_api_v2 python=3.8 阅读全文
posted @ 2024-05-07 19:35 张幼安 阅读(14) 评论(0) 推荐(0) 编辑
摘要: 解决方法:通过修改VSCode的launch.json文件。 参考: 用vscode调试python程序与launch.json的修改 - 知乎 (zhihu.com) 解决vscode在库里无法打断点问题 - 知乎 (zhihu.com) 问题描述:debug无法进入封装成package的包内。 阅读全文
posted @ 2024-04-15 19:49 张幼安 阅读(90) 评论(0) 推荐(0) 编辑
摘要: 上述为按照yolo v8 pose官方教程写的代码。 使用了python3.7, 以及对应的torch版本如下后,报错。AttributeError: module ‘torch‘ has no attribute ‘_six‘ 解决方法:更换了虚拟环境,重新安装yltralytics. 新环境: 阅读全文
posted @ 2024-04-10 10:40 张幼安 阅读(82) 评论(0) 推荐(0) 编辑
摘要: 1. 问题:使用yolo的时候, from ultralytics import YOLO. 然后报错: libcublas.so.11: symbol cublasLtGetStatusString version libcublasLt.so.11 not defined in file lib 阅读全文
posted @ 2024-04-10 10:18 张幼安 阅读(406) 评论(0) 推荐(0) 编辑