- 博客(12)
- 收藏
- 关注
RSS订阅原创 pytorch 打印参数
**遍历named_parameters()中的所有的参数，只打印那些param.requires_grad=True的变量**for name, param in model.named_parameters(): if param.requires_grad: print(name)若想可视化某一层的参数，可以先转列表，然后索引 params = list...
2019-11-12 20:32:37
445
原创 tensorboardx当网络模型输入两个参数时add_graph()报错解决方法
pytorch:0.4.0python:3.6File "/user/files/apps/anaconda2/envs/pytorch/lib/python3.6/site-packages/tensorboardX/graph.py", line 85, in graphlist_of_nodes = parse(graph)File "/user/files/apps/anacond...
2019-11-09 16:47:31
3186
原创 DataLossError (see above for traceback): Unable to open table file原因总结
如保存操作为saver.save(sess, “train_model”)那么会产生四个文件，这是python3的saver.restore()格式，分别为：1.checkpoint2.train_model-100.index3.train_model-100.meta4.train_model-100.data-00000-of-00001那么导入时，saver.restore(...
2019-10-24 10:38:44
1267
原创 NVIDA driver更新，Failed to initialize NVML: Driver/library version mismatch的问题
直接重启就好！
2019-10-24 09:59:58
1133
原创 tensorflow tensor可视化
tensorflow里可视化可分为两种：tensorflow的计算过程都是以tensor的形式输出的, 需要想本法从tensor转换成np.ndarray类型, 然后再使用 openCV,pandas,matplotlib等绘图工具库来绘制图像.将tensor–>np.ndarray只需调用sess.run(tensor)即可;接下来是使用绘图工具绘制图片了,plt提供了对矩阵的绘制绘...
2019-10-17 15:39:08
1760
原创 tensorflow中使用tf.ConfigProto()配置Session运行参数
tf.ConfigProto()函数用在创建session的时候，用来对session进行参数配置：config = tf.ConfigProto(allow_soft_placement=True, allow_soft_placement=True) config.gpu_options.per_process_gpu_memory_fraction = 0.4 #占用40%显存 ...
2019-09-28 02:15:41
534
转载 RCAN, EDSR代码dataloader.py出现错误
环境：python3.7+ cuda10.0 + pytorch1.1.0在运行MSRN代码时出现老多问题了，解决的路上全是心酸泪我的妈呀！在此记录下来，希望朋友们少走点弯路【花花问题：ImportError: cannot import name ‘_update_worker_pids’原因：MSRN中的dataloader文件中会出现很多这类问题，找不到对应的函数名和属性名等等，...
2019-09-25 08:59:32
2250
2
原创 代码复现中遇到的问题
Q：TypeError: Expected int32, got list containing Tensors of type '_Message' instead.A：tensorflow tf.concat的版本问题net = tf.concat(3, [branch_0, branch_1, branch_2, branch_3])should be changed tonet...
2019-09-07 10:38:17
653
原创 tensorflow中 tf.ConfigProto()的用法
控制代码对显存的使用：config = tf.ConfigProto(allow_soft_placement=True, allow_soft_placement=True)config.gpu_options.per_process_gpu_memory_fraction = 0.4 #占用40%显存sess = tf.Session(config=config)或者gpu_...
2019-08-29 11:15:37
365
原创 conda channel的镜像设置（设置可用清华源）
这里写自定义目录标题conda config --show能够显示出所有conda的config信息如果我们只想看channels的信息，输入conda config --show channels即可移除原来的没用的源eg:conda config --remove channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/...
2019-08-26 21:36:44
1537
原创 tensrflow基础学习
tensorflow基础学习欢迎使用Markdown编辑器保存与恢复模型及变量保存#Create some variables.v1 = tf.get_variable(“v1”, shape=3, initializer = tf.zeros_initializer)v2 = tf.get_variable(“v2”, shape=[5], initializer = tf.zero...
2019-08-25 09:55:06
162
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人