提前定义好模型,输出到txt的代码如下:
n_parameters = sum(p.numel() for p in model.parameters() if p.requires_grad)
print('number of params:', n_parameters)
modelsave = 'modeloutput.txt'
with open(modelsave, mode='w') as f:
f.write('\nThe model is:\n')
f.write(str(model)) # 此即上边定义的模型
f.write(str('\nTotal number of parameters:{}\n'.format(n_parameters)))
f.close()
print('model:', model)
print('numberofparams:', n_parameters)