in models/densenet_efficient_multi_gpu.py [0:0]
def forward(self, x):
if isinstance(x, Variable):
prev_features = [x]
else:
prev_features = x
out = self.efficient(prev_features)
# out = self.conv1(out)
if self.bn_size > 0:
out = self.bn(out)
out = self.relu(out)
out = self.conv(out)
return out