sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-26 02:43:02 +00:00
Code Issues Projects Releases Packages Wiki Activity

debug

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-11-24 18:24:03 +00:00
parent 1f60f2a580
commit d55a13f536
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
aimodel/src/subcommands/train_mono_predict.py
View file

@ -149,7 +149,10 @@ def do_jsonl(args, ai, dataset, model_params, do_argmax=False):
for batch in batched_iterator(dataset, tensors_in_item=2, batch_size=model_params["batch_size"]): for batch in batched_iterator(dataset, tensors_in_item=2, batch_size=model_params["batch_size"]):
rainfall_actual_batch, water_actual_batch = batch rainfall_actual_batch, water_actual_batch = batch
print("DEBUG:do_jsonl rainfall_actual_batch", rainfall_actual_batch.shape)
print("DEBUG:do_jsonl water_actual_batch", water_actual_batch.shape)
water_predict_batch = ai.embed(rainfall_actual_batch) water_predict_batch = ai.embed(rainfall_actual_batch)
print("DEBUG:do_jsonl water_actual_batch", water_predict_batch.shape)
water_actual_batch = tf.unstack(water_actual_batch, axis=0) water_actual_batch = tf.unstack(water_actual_batch, axis=0)
rainfall_actual_batch = tf.unstack(rainfall_actual_batch, axis=0) rainfall_actual_batch = tf.unstack(rainfall_actual_batch, axis=0)