From 70ae3e1608ad8f079761d42c22af736a7798b5f6 Mon Sep 17 00:00:00 2001 From: Salem Date: Thu, 21 Mar 2024 20:35:01 +0400 Subject: [PATCH] revert back to default TB in script --- tutorials/examples/train_hypergrid.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorials/examples/train_hypergrid.py b/tutorials/examples/train_hypergrid.py index 97157880..e75357fc 100644 --- a/tutorials/examples/train_hypergrid.py +++ b/tutorials/examples/train_hypergrid.py @@ -297,7 +297,7 @@ def main(args): # noqa: C901 "--loss", type=str, choices=["FM", "TB", "DB", "SubTB", "ZVar", "ModifiedDB"], - default="FM", + default="TB", help="Loss function to use", ) parser.add_argument(