要打印PyTorch模型的参数,可以使用以下代码:
import torch
# 定义模型
model = YourModel()
# 打印模型的参数
for name, param in model.named_parameters():
if param.requires_grad:
print(name, param.data)
这段代码将打印出模型的每个参数的名称和对应的数值。您可以根据需要对打印的内容进行进一步的格式化或处理。
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
推荐阅读:tensorflow怎么打印模型结构