pytorch如何知道某个Parameter是在哪一个Module中的创建的

最新推荐文章于 2025-08-20 09:14:45 发布

Hi20240217

最新推荐文章于 2025-08-20 09:14:45 发布

阅读量543

点赞数 10

CC 4.0 BY-SA版权

分类专栏：调试工具文章标签： pytorch python 人工智能

本文链接：https://blog.csdn.net/m0_61864577/article/details/138748564

pytorch如何知道某个Parameter是在哪一个Module中的创建的

代码

在定位pytorch精度问题时,发现optimizer中某些Parameter值异常,想知道它属于哪个模块的.本文提供二种方法
1.全局搜索
2.在创建Parameter的地方加一个属性,写明所在的模块名,需要的时候直接获取

代码

import torch
import sys
sys.setrecursionlimit(1000)
        
def search_recursive(var,stack,_id,depth):
    if var.__class__.__name__ in [
                                    "module","type","NoneType",
                                    "str","int","function","method-wrapper",
                                    "builtin_function_or_method",
                                    "method","_TensorMeta",
                                    "Tensor","method_descriptor",
                                    "bool","device","dtype",
                                    "getset_descriptor","layout",
                                    "wrapper_descriptor","property",
                                    "_ParameterM