import torch
class AE_LSTM(nn.Module):
def init(self, in_channel=1):
super(AE_LSTM, self).init()
self.encoder = nn.Sequential( nn.LSTM(in_channel, 16, batch_first=True, bidirectional=True), nn.InstanceNorm1d(16),
nn.ReLU(),
nn
BUG修复:__init__() got an unexpected keyword argument ‘in_channel‘
最新推荐文章于 2024-05-09 10:32:01 发布