摘要: 跑模型的时候出现了类似的错误: 错误:无法将3当作一个数据类型,那么我们可以知道肯定是缺少了参数,查看np.ones的参数如下: 改为这样就可以了: c = np.ones((2,3,4)) 阅读全文
posted @ 2021-10-13 22:29 Tomorrow1126 阅读(1161) 评论(0) 推荐(0) 编辑
摘要: 1、确保tensorflow是gpu版本的 import tensorflow as tf print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU'))) 运行结果:Num GPUs Av 阅读全文
posted @ 2021-10-13 16:00 Tomorrow1126 阅读(495) 评论(0) 推荐(0) 编辑
摘要: 问题是导入的模块问题 比如导入tqdm,不能用import tqdm,应该用from tqdm import tqdm 阅读全文
posted @ 2021-10-13 15:53 Tomorrow1126 阅读(300) 评论(0) 推荐(0) 编辑
摘要: #method1 sys import sys gpus = sys.argv[1] batch_size = sys.argv[2] print(gpus) print(batch_size) #method2 argparse 输入参数时要输入 -gpu=1 import argparse pa 阅读全文
posted @ 2021-10-13 15:50 Tomorrow1126 阅读(255) 评论(0) 推荐(0) 编辑
摘要: 解决办法 安装gast库 pip install gast==0.2.2 安装之后,这些warning就消失了 参考:https://github.com/youdao-ai/SRNet/issues/4 链接:https://blog.csdn.net/sinat_29957455/article 阅读全文
posted @ 2021-10-13 15:40 Tomorrow1126 阅读(244) 评论(0) 推荐(0) 编辑
摘要: 增加一个维度 import torch as t a = t.randn((128,)) a.shape >>>torch.Size([128]) a[:, None].shape >>>torch.Size([128, 1]) 阅读全文
posted @ 2021-10-13 15:38 Tomorrow1126 阅读(100) 评论(0) 推荐(0) 编辑
摘要: plt.imshow()函数负责对图像进行处理,并显示其格式,但是不能显示。 其后跟着plt.show()才能显示出来。 plt.imshow()用来绘制热图 链接:https://www.jb51.net/article/193182.htm https://blog.csdn.net/weixi 阅读全文
posted @ 2021-10-13 15:27 Tomorrow1126 阅读(1213) 评论(0) 推荐(0) 编辑
摘要: 昨天跑模型,发现测试的时候出了一大堆的warning,于是查了一下如何屏蔽日志信息: os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' tf.logging.set_verbosity(tf.logging.ERROR) 出现一些Warning等等的日志信息是因为Te 阅读全文
posted @ 2021-10-13 15:25 Tomorrow1126 阅读(229) 评论(0) 推荐(0) 编辑
摘要: 今天师兄跟我说跑h-lstm模型中fps逐渐降低,考虑是内存的问题 然后看了一眼代码,我考虑到是plt.show的问题,后来仔细一想,又对照了上次改的代码,发现没写plt.close() 写上以后就解决了...以后不要犯低级错误了 阅读全文
posted @ 2021-10-13 15:17 Tomorrow1126 阅读(41) 评论(0) 推荐(0) 编辑