要打印PyTorch模型的参数,可以使用以下代码:
import torch # 定义模型 model = YourModel() # 打印模型的参数 for name, param in model.named_parameters(): if param.requires_grad: print(name, param.data)
这段代码将打印出模型的每个参数的名称和对应的数值。您可以根据需要对打印的内容进行进一步的格式化或处理。