revert back to default TB in script

GFNOrg · Mar 21, 2024 · 70ae3e1 · 70ae3e1
1 parent 32ba79b
commit 70ae3e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(