随笔分类 -  AI part

摘要: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 张幼安 阅读(36) 评论(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 张幼安 阅读(162) 评论(0) 推荐(0) 编辑
摘要:做模型部署边缘设备的时候,我们经常会遇到特定格式的要求。但常见的onnx2tf很多时候都不能满足我们的要求。因此,记录一下我的操作过程。 1. 环境:(linux18.04) # Name Version Build Channel _libgcc_mutex 0.1 main defaults _ 阅读全文
posted @ 2024-06-05 14:11 张幼安 阅读(676) 评论(0) 推荐(0) 编辑
摘要:背景如下:我要删除Round节点, 同时看了一下,Dequantize和Quantize也是没有必要的。所以最好一起删除。 原始项目地址:PINTO0309/hand-gesture-recognition-using-onnx: This is a hand gesture recognition 阅读全文
posted @ 2024-05-30 20:43 张幼安 阅读(31) 评论(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 张幼安 阅读(21) 评论(0) 推荐(0) 编辑
摘要:上述为按照yolo v8 pose官方教程写的代码。 使用了python3.7, 以及对应的torch版本如下后,报错。AttributeError: module ‘torch‘ has no attribute ‘_six‘ 解决方法:更换了虚拟环境,重新安装yltralytics. 新环境: 阅读全文
posted @ 2024-04-10 10:40 张幼安 阅读(91) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示