Skip to content

GitLab

"docs/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "1398540ede05147a79e5d05aa8abc79fd30b688e"
Unverified Commit 7eaf9c24 authored by Yuge Zhang's avatar Yuge Zhang Committed by GitHub
Browse files

Add start_epoch configuration in PFLD example (#3709)

parent e67fea2c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
examples/nas/oneshot/pfld/train.py
View file @ 7eaf9c24
...@@ -63,6 +63,7 @@ def main(args): ...@@ -63,6 +63,7 @@ def main(args):
alpha=args.alpha, alpha=args.alpha,
beta=args.beta, beta=args.beta,
search_space=search_space, search_space=search_space,
start_epoch=args.start_epoch,
) )
# look-up table with information of search space, flops per block, etc. # look-up table with information of search space, flops per block, etc.
lookup_table = LookUpTable(config=nas_config, primitives=PRIMITIVES) lookup_table = LookUpTable(config=nas_config, primitives=PRIMITIVES)
...@@ -162,6 +163,7 @@ def parse_args(): ...@@ -162,6 +163,7 @@ def parse_args():
) )
parser.add_argument("--alpha", default=0.25, type=float) parser.add_argument("--alpha", default=0.25, type=float)
parser.add_argument("--beta", default=0.6, type=float) parser.add_argument("--beta", default=0.6, type=float)
parser.add_argument("--start_epoch", default=50, type=int)
parser.add_argument("--end_epoch", default=300, type=int) parser.add_argument("--end_epoch", default=300, type=int)
parser.add_argument( parser.add_argument(
"--snapshot", default="models", type=str, metavar="PATH" "--snapshot", default="models", type=str, metavar="PATH"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment