Module.named_parameters()、Module.named_children()與Module.named_modules()的區別

在代碼裏面輸出一會兒:python import torch import torch.nn as nn class Net(nn.Module): def __init__(self): super().__init__() self.fc1 = nn.Linear(3, 4) self.relu1 = nn.ReLU()
相關文章
相關標籤/搜索