From 1103ae5561b719f6cf9699402cce93ad1fb37b6c Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 16 Sep 2022 16:07:16 +0100 Subject: [PATCH] ai: tweak progress bar --- aimodel/src/subcommands/pretrain_predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/pretrain_predict.py b/aimodel/src/subcommands/pretrain_predict.py index 62a74f5..fadddf8 100644 --- a/aimodel/src/subcommands/pretrain_predict.py +++ b/aimodel/src/subcommands/pretrain_predict.py @@ -70,7 +70,7 @@ def run(args): for rainfall in ai.embed(dataset): handle.write(json.dumps(rainfall.numpy().tolist(), separators=(',', ':'))+"\n") # Ref https://stackoverflow.com/a/64710892/1460422 - if i == 0 or i % 1000 == 0: + if i == 0 or i % 100 == 0: sys.stderr.write(f"[pretrain:predict] STEP {i}\r") i += 1