Merge pull request NVlabs#3 from pbizimis/main

dvschultz · web-flow · commit 182a62cbc011 · 2021-03-24T01:58:22.000-04:00
Add weight layer to cff
diff --git a/closed_form_factorization.py b/closed_form_factorization.py
@@ -21,7 +21,7 @@
     modulate = {
         k[0]: k[1]
         for k in G.named_parameters()
-        if "affine" in k[0] and "torgb" not in k[0] and "weight" in k[0]
+        if "affine" in k[0] and "torgb" not in k[0] and "weight" in k[0] or ("torgb" in k[0] and "b4" in k[0] and "weight" in k[0] and "affine" in k[0])
     }
 
     weight_mat = []

Original file line number	Diff line number	Diff line change
`@@ -21,7 +21,7 @@`
`21`	`21`	`modulate = {`
`22`	`22`	`k[0]: k[1]`
`23`	`23`	`for k in G.named_parameters()`
`24`		`- if "affine" in k[0] and "torgb" not in k[0] and "weight" in k[0]`
	`24`	`+ if "affine" in k[0] and "torgb" not in k[0] and "weight" in k[0] or ("torgb" in k[0] and "b4" in k[0] and "weight" in k[0] and "affine" in k[0])`
`25`	`25`	`}`
`26`	`26`
`27`	`27`	`weight_mat = []`