mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-04 17:13:02 +00:00
debug2
This commit is contained in:
parent
d55a13f536
commit
3131b4f7b3
1 changed files with 1 additions and 1 deletions
|
@ -152,12 +152,12 @@ def do_jsonl(args, ai, dataset, model_params, do_argmax=False):
|
|||
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)
|
||||
|
||||
i_batch = 0
|
||||
for water_predict in water_predict_batch:
|
||||
print("DEBUG:do_jsonl water_predict", water_predict.shape)
|
||||
# [ width, height, softmax_probabilities ] → [ batch, width, height ]
|
||||
if do_argmax:
|
||||
water_predict = tf.math.argmax(water_predict, axis=-1)
|
||||
|
|
Loading…
Reference in a new issue