摘要：
1、确保tensorflow是gpu版本的 import tensorflow as tf print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU'))) 运行结果：Num GPUs Av 阅读全文
摘要：
问题是导入的模块问题 比如导入tqdm，不能用import tqdm，应该用from tqdm import tqdm 阅读全文
摘要：
#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 阅读全文
摘要：
解决办法 安装gast库 pip install gast==0.2.2 安装之后，这些warning就消失了 参考：https://github.com/youdao-ai/SRNet/issues/4 链接：https://blog.csdn.net/sinat_29957455/article 阅读全文
摘要：
增加一个维度 import torch as t a = t.randn((128,)) a.shape >>>torch.Size([128]) a[:, None].shape >>>torch.Size([128, 1]) 阅读全文
摘要：
plt.imshow()函数负责对图像进行处理，并显示其格式，但是不能显示。 其后跟着plt.show()才能显示出来。 plt.imshow()用来绘制热图 链接：https://www.jb51.net/article/193182.htm https://blog.csdn.net/weixi 阅读全文
摘要：
昨天跑模型，发现测试的时候出了一大堆的warning，于是查了一下如何屏蔽日志信息： os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' tf.logging.set_verbosity(tf.logging.ERROR) 出现一些Warning等等的日志信息是因为Te 阅读全文
摘要：
今天师兄跟我说跑h-lstm模型中fps逐渐降低，考虑是内存的问题 然后看了一眼代码，我考虑到是plt.show的问题，后来仔细一想，又对照了上次改的代码，发现没写plt.close() 写上以后就解决了...以后不要犯低级错误了 阅读全文