Skip to content
Snippets Groups Projects
Commit e8d17a34 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 4fd818b0
Branches
No related tags found
No related merge requests found
......@@ -1199,6 +1199,9 @@ def run_average_models(ckpt_dir_s_path, day_night='NIGHT', l1b_andor_l2='BOTH',
model_lyrs[k].append(w)
for lyr in model_lyrs:
nda = np.stack(lyr)
print(nda.shape)
avg = np.mean(nda)
print(avg.shape)
avg_model_weights.append(nda)
# -- Make a new model for the averaged weights
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment