weixin_41838969的博客

原创 直接输出深度学习模型和参数的代码

提前定义好模型，输出到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.