Spaces:
mashroo
/
Running on Zero

YoussefAnso commited on
Commit
96bbb06
·
1 Parent(s): 00c2ac2

Add newline at the end of mlp_head.py for consistency and to adhere to coding standards

Browse files
Files changed (1) hide show
  1. model/archs/mlp_head.py +1 -0
model/archs/mlp_head.py CHANGED
@@ -36,6 +36,7 @@ class RgbMlp(nn.Module):
36
  x = F.relu(self.fc2(x))
37
  out = self.fc3(x)
38
  out = torch.tanh(out)
 
39
  return out
40
 
41
 
 
36
  x = F.relu(self.fc2(x))
37
  out = self.fc3(x)
38
  out = torch.tanh(out)
39
+
40
  return out
41
 
42