core.nn.nsf.create_nsf_with_rb_projection_embedding_net

core.nn.nsf.create_nsf_with_rb_projection_embedding_net(
    posterior_kwargs,
    embedding_kwargs,
    initial_weights=None,
)

Builds a neural spline flow with an embedding network that consists of a reduced basis projection followed by a residual network. Optionally initializes the embedding network weights.

Parameters

Name	Type	Description	Default
posterior_kwargs	dict	kwargs for neural spline flow	required
embedding_kwargs	dict	kwargs for emebedding network	required
initial_weights	dict	Dictionary containing the initial weights for the SVD projection. This should have one key ‘V_rb_list’, with value a list of SVD V matrices (one for each detector).	None

Returns

Name	Type	Description
	nn.Module	Neural spline flow model