【pytorch与tensorflow框架】—深度学习中神经网络结构查看/打印
一、pytorch打印网络结构
#前提:构建了一个TransformerModel网络结构类
#实例化网络对象
model = TransformerModel(ntokens, emsize, nhead, nhid, nlayers, dropout)
法一:主要查看网络层次结构,也包括输入输出参数
print(model)
结果如下:
TransformerModel(
(pos_encoder): PositionalEncoding(
(dropout): Dropout(
原创
2020-09-20 16:06:00 ·
1237 阅读 ·
0 评论