Pytorch 查看網絡模型參數

能夠經過model.state_dict()或者model.named_parameters()函數查看如今的所有可訓練參數(包括經過繼承獲得的父類中的參數) 代碼以下:python import torch # 引入torch.nn並指定別名 import torch.nn as nn import torch.nn.functional as F class Net(nn.Module):
相關文章
相關標籤/搜索