mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-04 17:13:02 +00:00
debug
This commit is contained in:
parent
1f60f2a580
commit
d55a13f536
1 changed files with 3 additions and 0 deletions
|
@ -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"]):
|
||||
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)
|
||||
print("DEBUG:do_jsonl water_actual_batch", water_predict_batch.shape)
|
||||
water_actual_batch = tf.unstack(water_actual_batch, axis=0)
|
||||
rainfall_actual_batch = tf.unstack(rainfall_actual_batch, axis=0)
|
||||
|
||||
|
|
Loading…
Reference in a new issue