import torch import torch.nn as nn class HumanoidSensorFusionModel(nn.Module): def __init__(self): super().__init__() self.fc1 = nn.Linear(320, 160) self.fc2 = nn.Linear(160, 64) def forward(self, x): x = torch.tanh(self.fc1(x)) return self.fc2(x)