fix: csd.py output order to match CSDOutput
Browse files
csd.py
CHANGED
@@ -41,4 +41,4 @@ class CSD(nn.Module):
|
|
41 |
content_output = features @ self.last_layer_content
|
42 |
content_output = F.normalize(content_output, dim=1, p=2)
|
43 |
|
44 |
-
return features,
|
|
|
41 |
content_output = features @ self.last_layer_content
|
42 |
content_output = F.normalize(content_output, dim=1, p=2)
|
43 |
|
44 |
+
return features, style_output, content_output
|