From a81f18e35be121431316bb64dfb1aa87ee02c729 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Wed, 18 Jan 2023 12:50:53 -0500 Subject: [PATCH] update defaults --- cli/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cli/train.py b/cli/train.py index c50953871..b8de87cda 100755 --- a/cli/train.py +++ b/cli/train.py @@ -366,7 +366,7 @@ async def main(): parser.add_argument("--vectors", type = int, default = -1, required = False, help = "number of vectors per token, default: dynamic based on number of input images") parser.add_argument("--batch", type = int, default = 1, required = False, help = "batch size, default: %(default)s") parser.add_argument("--rate", type = str, default = "", required = False, help = "learning rate, default: dynamic") - parser.add_argument("--rstart", type = float, default = 0.05, required = False, help = "starting learn rate if using dynamic rate, default: %(default)s") + parser.add_argument("--rstart", type = float, default = 0.02, required = False, help = "starting learn rate if using dynamic rate, default: %(default)s") parser.add_argument("--rend", type = float, default = 0.0001, required = False, help = "ending learn rate if using dynamic rate, default: %(default)s") parser.add_argument("--rdescend", type = float, default = 2, required = False, help = "learn rate descend power when using dynamic rate, default: %(default)s") parser.add_argument("--grad", type = int, default = 20, required = False, help = "accumulate gradient over n images, default: : %(default)s")