输出网络参数和网络结构#

1
print("\n\nprint the each layer's name(GAB)")
2
for name in G_AB.state_dict():
3
  print(name)
4
print("\n\nprint the name and the size of layers")
5
for name,parameters in G_AB.named_parameters():
6
  print(name,";",parameters.size())
7
print("\n\nsave the parameters")
8
for layer in ['model.1.freq','model.1.theta','model.1.sigma','model.1.psi','model.1.x','model.1.y','model.1.weight']:
9
  print("\n",layer)
10
  print(G_AB.state_dict()[layer])
11
  G_AB.state_dict()[layer].cpu().detach().numpy().tofile(layer[8:]+'.rsf@')

1
from torchinfo import summary
2
summary(G_AB,input_size=(16,1,256,256))