"tools/imglab/vscode:/vscode.git/clone" did not exist on "b090ef89ef3dd3ab021c8ff855669d4d8bf68fcb"
Commit b30cf9d2 authored by Chengjiang Long's avatar Chengjiang Long Committed by Facebook GitHub Bot
Browse files

Implement the user calibration model option 3 under D2GO

Summary:
Dataloader:
Rewrote the data loader via build_stream_dataset_reader with the DATASET_DEFINITION of "peopleai_face_eng_inference_results".

User Calibration Model (initial version):

nn.Sequential(
    nn.Conv1d(72, 128, 1),
    nn.BatchNorm1d(128),
    nn.ReLU(),
    nn.Flatten(),
    nn.Linear(128, 72),
)

Differential Revision: D34202009

fbshipit-source-id: 55a2c579e463ed19eac38b5dd12e11c09cbccc11
parent f781223c
...@@ -161,10 +161,11 @@ def prepare_for_launch(args): ...@@ -161,10 +161,11 @@ def prepare_for_launch(args):
- returned output_dir will always be non empty, args.output_dir has higher - returned output_dir will always be non empty, args.output_dir has higher
priority than cfg.OUTPUT_DIR. priority than cfg.OUTPUT_DIR.
""" """
print(args) logger.info(args)
runner = create_runner(args.runner) runner = create_runner(args.runner)
cfg = runner.get_default_cfg() cfg = runner.get_default_cfg()
if args.config_file: if args.config_file:
with PathManager.open(reroute_config_path(args.config_file), "r") as f: with PathManager.open(reroute_config_path(args.config_file), "r") as f:
print("Loaded config file {}:\n{}".format(args.config_file, f.read())) print("Loaded config file {}:\n{}".format(args.config_file, f.read()))
......
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