- 博客(2)
- 收藏
- 关注
原创 使用pytorch创建resnet模型
使用pytorch创建resnet模型环境 win11 rtx3060anaconda 2020.07tensorflow 2.4.1cuda 11.1cndnn 8.0.5python 3.8pytorch 1.8tensorflow-addons 0.15.0pandas 1.3.5seaborn 0.11.2numpy 1.19.5pytorch=1.8导入数据https://blog.csdn.net/qq_24193303/article/details/1081944
2022-01-14 13:12:13 428
原创 session.run运行速度过慢的问题
最近在运行数据增强算法,跑着跑着就跑不动了。代码本身也没有错误,就是运行速度太慢,现在对代码进行优化。1.考虑是图节点持续增多在运行session.run的过程中,会持续增加节点,这样就会让整张图越来越大,速度也会越来越慢。考虑 每次运行都会重置这张图,这样节点就不会增加了。tensorflow.compat.v1.reset_default_graph() # 用于清除默认图形堆栈并重置全局默认图形`上述代码放在建图之前,就是在用占位符建立图之前使用。tensorflow.compat.v
2021-12-28 11:15:06 1306
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人