def _add_args(p):
p.add_argument("--steps",
help="number of steps for recurrent models",
type=int,
default=5,
required=False)
AdjustableSigmaMixin._add_args(p)
def _add_args(p):
p.add_argument("--npoints",
help="number of control points",
type=int,
default=16,
required=False)
p.add_argument("--nlines",
help="number of independent lines",
type=int,
default=1,
required=False)
p.add_argument("--decoder-hidden",
help="decoder hidden size",
type=int,
default=64,
required=False)
p.add_argument("--decoder-hidden2",
help="decoder hidden2 size",
type=int,
default=256,
required=False)
p.add_argument("--edt-approximation",
help="approximation to use for computing edt",
type=str,
default="polyline",
required=False,
choices=['polyline', 'bruteforce'])
AdjustableSigmaMixin._add_args(p)
def _add_args(p):
p.add_argument("--npoints",
help="number of points",
type=int,
default=50,
required=False)
p.add_argument("--decoder-hidden",
help="decoder hidden size",
type=int,
default=64,
required=False)
p.add_argument("--decoder-hidden2",
help="decoder hidden2 size",
type=int,
default=256,
required=False)
AdjustableSigmaMixin._add_args(p)