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

flip squeeze operator ordering

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-10-21 15:38:57 +01:00
parent af26964c6a
commit bb0258f5cd
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/subcommands/train_predict.py
View file

@ -99,7 +99,7 @@ def do_png(args, ai, dataset, model_code):
rainfall, water = item
water_predict_batch = ai.embed(rainfall)
water = tf.unstack(tf.squeeze(water), axis=0)
water = tf.squeeze(tf.unstack(water, axis=0))
i_batch = 0
for water_predict in water_predict_batch: