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
|