第1关:网络层

import torch.nn as nn

myNet = nn.Sequential(
    ########## Begin ##########
    nn.Linear(3, 100),
   
    nn.Linear(100, 1000),

    nn.Linear(1000, 100),

    nn.Linear(100, 10)
)
    ########## End ##########
print(myNet)

第2关:激活函数

import torch
from torch.autograd import Variable
import torch.nn.functional as F

x = torch.linspace(-5, 5, 200)   # 构造一段连续的数据
x = Variable(x) 

########## Begin ##########

# 使用sigmoid激活函数
y_sigmoid = torch.sigmoid(x)
print("y_sigmoid:")
print(y_sigmoid)

# 使用tanh激活函数
y_tanh = torch.tanh(x)
print("y_tanh:")
print(y_tanh)

# 使用ReLU激活函数
y_relu = torch.relu(x)
print("y_relu:")
print(y_relu)

# 使用softplus激活函数
y_softplus = F.softplus(x)
print("y_softplus:")
print(y_softplus)

# 使用leakyReLU激活函数
y_leakyrelu = F.leaky_relu(x)
print("y_leakyrelu:")
print(y_leakyrelu)

# 使用ELU激活函数
y_ELU = F.elu(x)
print("y_ELU:")
print(y_ELU)

# 使用SELU激活函数
y_SELU = F.selu(x)
print("y_SELU:")
print(y_SELU)
########## End ##########

第3关:模型优化

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐