From c78b6562ef9eeb7e536224aa43bc3cc214803bc9 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 2 Sep 2022 17:00:21 +0100 Subject: [PATCH] debug adjustment --- aimodel/src/subcommands/pretrain.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/aimodel/src/subcommands/pretrain.py b/aimodel/src/subcommands/pretrain.py index 05787f0..d1e2737 100644 --- a/aimodel/src/subcommands/pretrain.py +++ b/aimodel/src/subcommands/pretrain.py @@ -39,11 +39,12 @@ def run(args): ) dataset_metadata = read_metadata(args.input) - # for items in dataset_train.repeat(10): - # print("ITEMS", len(items)) - # print("LEFT", [ item.shape for item in items[0] ]) + # for (items, label) in dataset_train: + # print("ITEMS", len(items), [ item.shape for item in items ]) + # print("LABEL", label.shape) # print("ITEMS DONE") # exit(0) + ai = RainfallWaterContraster( dir_output=args.output,