def __init__(self, h):
super().__init__()
self.conv1 = tu.Conv2d(8, int(h / 2), 3, stride=2)
self.conv2 = tu.Conv2d(int(h / 2), h, 3, stride=2)
self.fc1 = torch.nn.Linear(6 + 4 * 4 * h + h, h)
# self.fc1 = torch.nn.Linear(5+4*4*h, h)
self.ent1 = torch.nn.Linear(6, h)
def __init__(self, h):
super().__init__()
self.conv1 = tu.Conv2d(8, h, 3, stride=2)
self.conv2 = tu.Conv2d(h, h, 3, stride=2)
self.fc1 = torch.nn.Linear(16 * h + 5, h)
def __init__(self, h):
super().__init__()
self.conv1 = tu.Conv2d(8, 4, 1)
self.fc1 = tu.FCRelu(5 + 900, h)