这是我按照你提供的建议改的代码我的return_all_layers 设置为 False但是我的输出outputs仍然是tensor-00146 -00113 -00137 00040 00058 00087 -00146 -00113 -00137 00040 00058 00087 -00174 -00124 -00138 -00201 -00219 -00215 -00174 -001
感谢您分享代码。在这个代码中,您已经将 return_all_layers 设置为 False 了,但是您的输出仍然是一个张量,说明您的模型仍然返回了所有的层的输出。这可能是因为您在循环中没有正确地返回最后一层的输出或者是因为您的模型定义中有一些问题。建议您检查一下模型定义中的代码,确保正确地返回了最后一层的输出。同时,您可以将 return_all_layers 设置为 True,这样可以查看所有层的输出,以确保模型的正确性。
原文地址: https://www.cveoy.top/t/topic/brY8 著作权归作者所有。请勿转载和采集!