-
Notifications
You must be signed in to change notification settings - Fork 2
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Do not merge] Extend post-processing and plotting to act on intermediate layers of model #64
base: master
Are you sure you want to change the base?
Conversation
Just the final commit that is |
This reverts commit f215adb. Reverting update to fix broken test, since made in other PR
Ok done. Didn't realise that was the breaking you were referring to. |
My fault for not being specific enough! If we get that merged soon then we can rebase this |
for block in loaded_model: | ||
v, _ = block(v, 0, negative_mag) | ||
# only want coupling layers | ||
if len([tensor for tensor in block.state_dict().values()]) > 1: | ||
if i == layer_id: | ||
return v | ||
else: | ||
i += 1 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I will need to adjust this after #68 is merged, since the model is no longer necessarily a flat Sequential
of coupling blocks.
Phase 1:
Phase 2:
layer_id
to allow looping the entire analysis (including correlation functions, bootstrapping etc) over each layer in the modelAmbitions:
layer_id
could refer to an attribute of the layers in the model. E.g., if I wanted to plot the spline layer and the MCMC output I could writelayer_ids: [spline, metropolis]
.