Skip to content
Snippets Groups Projects
Commit cf4ecd32 authored by Julius Richter's avatar Julius Richter
Browse files

centered=True as default

parent 72096667
No related branches found
No related tags found
No related merge requests found
...@@ -63,7 +63,7 @@ class NCSNpp(nn.Module): ...@@ -63,7 +63,7 @@ class NCSNpp(nn.Module):
image_size = 256, image_size = 256,
embedding_type = 'fourier', embedding_type = 'fourier',
dropout = .0, dropout = .0,
centered = False, centered = True,
**unused_kwargs **unused_kwargs
): ):
super().__init__() super().__init__()
...@@ -249,7 +249,9 @@ class NCSNpp(nn.Module): ...@@ -249,7 +249,9 @@ class NCSNpp(nn.Module):
@staticmethod @staticmethod
def add_argparse_args(parser): def add_argparse_args(parser):
parser.add_argument("--centered", action="store_true", help="The data is already centered [-1, 1]") 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.set_defaults(centered=True)
return parser return parser
def forward(self, x, time_cond): def forward(self, x, time_cond):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment