printing final predictions by default to False

This commit is contained in:
svlandeg 2020-07-31 17:36:32 +02:00
parent cc2f58a1b0
commit 878327d38e
1 changed files with 1 additions and 1 deletions

View File

@ -25,7 +25,7 @@ def debug_model_cli(
P0: bool = Opt(False, "--print-step0", "-P0", help="Print model before training"),
P1: bool = Opt(False, "--print-step1", "-P1", help="Print model after initialization"),
P2: bool = Opt(False, "--print-step2", "-P2", help="Print model after training"),
P3: bool = Opt(True, "--print-step3", "-P3", help="Print final predictions"),
P3: bool = Opt(False, "--print-step3", "-P3", help="Print final predictions"),
use_gpu: int = Opt(-1, "--gpu-id", "-g", help="GPU ID or -1 for CPU")
# fmt: on
):