pytorch如何打印模型參數(shù)

小億
246
2024-03-29 13:11:56

要打印PyTorch模型的參數(shù)，可以使用以下代碼：

import torch

# 定義模型
model = YourModel()

# 打印模型的參數(shù)
for name, param in model.named_parameters():
    if param.requires_grad:
        print(name, param.data)

這段代碼將打印出模型的每個(gè)參數(shù)的名稱和對(duì)應(yīng)的數(shù)值。您可以根據(jù)需要對(duì)打印的內(nèi)容進(jìn)行進(jìn)一步的格式化或處理。

0 踩