model init 结束以后就可以 查看模型参数了
1
2
3
4
5
6
7
8
9
|
model = Model(args)
summary(model, input_data=[x_enc, x_mark_enc, x_dec, x_mark_dec])
print(model)
total_params = sum(p.numel() for p in model.parameters())
print(f"模型总参数量: {total_params:,}")
outputs = model(x_enc,x_mark_enc,x_dec,x_mark_dec)
print(outputs.shape)
|