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- 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 |
|