sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-22 09:13:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

ai: tweak progress bar

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-16 16:07:16 +01:00
parent 1e35802d2b
commit 1103ae5561
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/subcommands/pretrain_predict.py
View file

@ -70,7 +70,7 @@ def run(args):
for rainfall in ai.embed(dataset): for rainfall in ai.embed(dataset):
handle.write(json.dumps(rainfall.numpy().tolist(), separators=(',', ':'))+"\n") # Ref https://stackoverflow.com/a/64710892/1460422 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") sys.stderr.write(f"[pretrain:predict] STEP {i}\r")
i += 1 i += 1