sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-25 10:32:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

moar debugging

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-10-31 18:26:34 +00:00
parent 172cf9d8ce
commit dfef7db421
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
aimodel/src/subcommands/pretrain.py
View file

@ -21,6 +21,13 @@ def parse_args():
return parser
def count_batches(dataset):
count = 0
for _ in dataset:
count += 1
return count
def run(args):
if (not hasattr(args, "water_size")) or args.water_size == None:
args.water_size = 100
@ -40,7 +47,11 @@ def run(args):
dirpath_input=args.input,
batch_size=args.batch_size,
)
dataset_metadata = read_metadata(args.input)
print("BATCHES_TRAIN", count_batches(dataset_train))
print("BATCHES_VALIDATE", count_batches(dataset_validate))
# for (items, label) in dataset_train:
# print("ITEMS", len(items), [ item.shape for item in items ])
@ -59,4 +70,3 @@ def run(args):
)
ai.train(dataset_train, dataset_validate)