Skip to content
Snippets Groups Projects
Commit 02bf1ec4 authored by jrichter's avatar jrichter
Browse files

remove double argument --attn_resolutions

parent 60d6fa09
No related branches found
No related tags found
No related merge requests found
......@@ -43,7 +43,6 @@ class NCSNpp_48k(nn.Module):
parser.add_argument("--num_res_blocks", type=int, default=2)
parser.add_argument("--attn_resolutions", type=int, nargs='+', default=[])
parser.add_argument("--nf", type=int, default=128, help="Number of channels to use in the model")
parser.add_argument("--attn_resolutions", type=int, nargs='+', default=[])
parser.add_argument("--no-centered", dest="centered", action="store_false", help="The data is not centered [-1, 1]")
parser.add_argument("--centered", dest="centered", action="store_true", help="The data is centered [-1, 1]")
parser.add_argument("--progressive", type=str, default='none', help="Progressive downsampling method")
......@@ -81,7 +80,7 @@ class NCSNpp_48k(nn.Module):
self.nf = nf = nf
ch_mult = ch_mult
self.num_res_blocks = num_res_blocks = num_res_blocks
self.attn_resolutions = attn_resolutions = attn_resolutions
self.attn_resolutions = attn_resolutions
dropout = dropout
resamp_with_conv = resamp_with_conv
self.num_resolutions = num_resolutions = len(ch_mult)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment