"git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "01e21e1283cb6a10805f0aeb1ae35bb6cd5b9dd4"
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
......@@ -63,6 +63,7 @@ def main(args):
alpha=args.alpha,
beta=args.beta,
search_space=search_space,
start_epoch=args.start_epoch,
)
# look-up table with information of search space, flops per block, etc.
lookup_table = LookUpTable(config=nas_config, primitives=PRIMITIVES)
......@@ -162,6 +163,7 @@ def parse_args():
)
parser.add_argument("--alpha", default=0.25, 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(
"--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