vvmatorin commited on
Commit
881be7a
·
verified ·
1 Parent(s): 6b50c70

fix: csd.py output order to match CSDOutput

Browse files
Files changed (1) hide show
  1. csd.py +1 -1
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, content_output, style_output
 
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