随笔分类 - tensorflow

关于tensorflow GPU训练Function call stack: train_function 问题

摘要：csdn 阅读全文

posted @ 2023-03-16 15:16 p_xcn 阅读(124) 评论(0) 推荐(0) 编辑

取消打印TensorFlow无用警告信息

摘要：取消打印CUDA信息以及CPU指令集不支持AVX2 import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' 取消打印Tensorflow 2.0API变更以及Pyinstaller打包为exe后额外增加的Warning import tensorflow 阅读全文

posted @ 2023-03-16 15:14 p_xcn 阅读(68) 评论(0) 推荐(0) 编辑

TensorFlow 的 Graph 模式转换

摘要：定义 TensorFlow 图形并将其保存到磁盘上。使用 TensorFlow 的 tf.Graph() 和 tf.Session() 函数来定义和运行 TensorFlow 图形，并使用 tf.train.write_graph() 函数将其保存到磁盘上。 import tensorflow a 阅读全文

posted @ 2023-02-23 14:20 p_xcn 阅读(55) 评论(0) 推荐(0) 编辑

TFlite 多线程并行

摘要：import numpy as np from multiprocessing import Process, Queue def process_data(data, model_TSNet, tsnet_input, tsnet_output, queue): model = tflite_ru 阅读全文

posted @ 2023-02-23 14:14 p_xcn 阅读(95) 评论(0) 推荐(0) 编辑

tensorflow2.0+TF-lite 各种报错

摘要：generic_type: type "InterpreterWrapper" is already registered! 原因：tensorflow2.5.0rc0版本太高，1.降低版本: pip install tensorflow==2.3 2.搜索整个代码库查找其他尝试注册"Interpr 阅读全文

posted @ 2023-02-22 17:39 p_xcn 阅读(402) 评论(0) 推荐(0) 编辑

-

公告

搜索

常用链接

随笔分类

随笔档案

阅读排行榜

随笔分类 - tensorflow