【tensorflow】变量的创建、初始化、保存和恢复

最新推荐文章于 2022-07-27 10:51:24 发布

大姨妈V

最新推荐文章于 2022-07-27 10:51:24 发布

阅读量1.5k

点赞数

分类专栏： tensorflow 从零开始学习卷积神经网络

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.youkuaiyun.com/u012679707/article/details/79912380

版权

本文介绍了如何在TensorFlow中管理变量，包括如何创建和初始化变量，以及如何保存和恢复模型的变量状态。通过`tf.Variable()`创建变量，并使用`tf.global_variables_initializer()`进行初始化。利用`tf.train.Saver()`对象实现模型的保存和恢复，可以选择保存和恢复特定的变量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【tensorflow】变量的创建、初始化、保存和恢复：

tf.Variable 类
tf.train.Saver 类

参考：

变量:创建、初始化、保存和加载 http://www.tensorfly.cn/tfdoc/how_tos/variables.html

<

最低0.47元/天解锁文章

博客等级

码龄12年

108
原创

1034
点赞

4962
收藏

859
粉丝

关注

私信

热门文章

分类专栏

最新评论

【机器学习】支持向量机SVM及实例应用
bo_hai: 楼主，你咋设置成vip可见呢？！你这水平，应该不差钱呀！
【翻译经典】Rapid Object Detection using aBoosted Cascade of Simple Features（人脸检测经典论文）
尧哥235: cvpr
三、生成RPM包
优快云-Ada助手: 不知道 CS入门技能树是否可以帮到你：https://edu.youkuaiyun.com/skill/gml?utm_source=AI_act_gml
三、生成RPM包
大姨妈V: ###### 1.生成可执行文件、库文件 ######### cmake_minimum_required(VERSION 3.15) project(compute) set(target zls_bin) set(target2 libcompute.so) # 依赖的头文件 include_directories(${PROJECT_SOURCE_DIR}/include) file(GLOB include_file "${PROJECT_SOURCE_DIR}/include/*.h") # 文件搜索 file(GLOB SRC "${PROJECT_SOURCE_DIR}/src/*.c") # 设置生成库文件的路径(静态库\动态库都可行) set(LIBRARY_OUTPUT_PATH ${PROJECT_SOURCE_DIR}/lib) # 设置生成可执行文件的路径 set(EXECUTABLE_OUTPUT_PATH ${PROJECT_SOURCE_DIR}/bin) set(CMAKE_VERBOSE_MAKEFILE ON) MESSAGE("PROJECT_NAME:${PROJECT_NAME}") MESSAGE("SRC:${SRC}") MESSAGE("LIBRARY_OUTPUT_PATH:${LIBRARY_OUTPUT_PATH}") MESSAGE("include_file:${include_file}") # 生成静态库文件 #add_library(${PROJECT_NAME} STATIC ${SRC}) # 生成动态库文件 add_library(${PROJECT_NAME} SHARED ${SRC}) # 生成可执行文件（方式三：链接动态库文件） ## 包含动态库路径 link_directories(${PROJECT_SOURCE_DIR}/lib) ## 生成可执行文件 add_executable(${target} main.c) ## 链接动态库文件(target 依赖动态库compute) target_link_libraries(${target} ${PROJECT_NAME})
三、生成RPM包
大姨妈V: # 将依赖的头文件安装到include目录 install(DIRECTORY ${PROJECT_SOURCE_DIR}/include DESTINATION include COMPONENT BComponent) #install(FILES ${include_file} DESTINATION include COMPONENT BComponent) include(CPack) # 添加一个名为 AComponent 的 component cpack_add_component(AComponent DISPLAY_NAME "A program" DESCRIPTION "The program for test" GROUP Aprogram) # 添加一个名为 BComponent 的 component cpack_add_component(BComponent DISPLAY_NAME "B program" DESCRIPTION "The program for test" GROUP Bprogram) # 添加一个名为 Aprogram 的 group, 这个名字会作为 rpm 包名字的一部分 cpack_add_component_group(Aprogram) # 添加一个名为 Bprogram 的 group cpack_add_component_group(Bprogram)

大家在看

最新文章

目录

评论 1

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。