关于深度学习的一些有用代码

计算模型的参数

### compute model params
def count_param(model):
    param_count = 0
    for param in model.parameters():
        param_count += param.view(-1).size()[0]
    return param_count

打印模型的信息

from torchinfo import summary
print(summary(model,x.shape))

画sigmoid函数曲线图

import numpy as np
import math
import matplotlib.pyplot as plt

x=np.arange(-10,10,0.1)
y=1/(1+np.exp(-x))
plt.plot(x,y,label='sigmoid')
plt.xlabel('x')
plt.ylabel('y')

plt.ylim(0,1)
plt.legend()
plt.show()


i