纪念自己第一次跑通CNN(结构化数据)

最新推荐文章于 2023-01-26 10:45:02 发布

林世东

最新推荐文章于 2023-01-26 10:45:02 发布

阅读量683

点赞数

文章标签：深度学习神经网络

本文链接：https://blog.csdn.net/lsdniu/article/details/118971138

版权

博主分享了自己首次成功运行卷积神经网络（CNN）的经历，解决了输入通道和输出通道的维度问题。文章提到了在(-1, x, 62)中设置通道数，并说明了线性层参数的调整方法，尤其是针对分类任务的参数设定。未来计划探索RNN和LSTM。" 127813610,11828448,Python面向对象编程基础,"['Python', '面向对象编程', '类', '继承']

摘要由CSDN通过智能技术生成

import torch
from torch import nn 
import importlib 
from torchkeras import Model,summary

class Net(Model):
    
    def __init__(self):
        super(Net, self).__init__()
        
        #设置padding_idx参数后将在训练过程中将填充的token始终赋值为0向量
        self.conv = nn.Sequential()
        self.conv.add_module("conv_1",nn.Conv1d(in_channels = 2,out_channels = 16,kernel_size = 5))
        self.conv.add_module(