8820d66965eeb2f5f7bfac58c0fcfc68e748d53a,pyprob/nn/inference_network.py,InferenceNetwork,_init_layers_observe_embedding,#InferenceNetwork#Any#Any#,64

Before Change

        observe_embedding_total_dim = 0
        for name, value in observe_embeddings.items():
            distribution = example_trace.named_variables[name].distribution
            if distribution is None:
                raise ValueError("Observable {}: cannot use this observation as an input to the inference network, because there is no associated likelihood.".format(name))
            else:
                if "reshape" in value:
                    input_shape = torch.Size(value["reshape"])
                else:
                    input_shape = distribution.sample().size()
            if "dim" in value:
                output_shape = torch.Size([value["dim"]])
            else:
                print("Observable {}: embedding dim not specified, using the default 256.".format(name))

After Change

                input_shape = torch.Size(value["reshape"])
            else:
                input_shape = variable.value.size()
            print("input_shape", input_shape)
            if "dim" in value:
                output_shape = torch.Size([value["dim"]])
            else:
                print("Observable {}: embedding dim not specified, using the default 256.".format(name))

Instances