a2954b5
1
2
3
4
5
6
7
8
9
10
11
12
13
import torch import torch.nn as nn import torch.nn.functional as F class SensorExpert(nn.Module): def __init__(self): super(SensorExpert, self).__init__() self.fc1 = nn.Linear(10, 128) def forward(self, x): x = F.relu(self.fc1(x)) return x
