MCPcopy Create free account

hub / github.com/addtt/ladder-vae-pytorch / functions

Functions88 in github.com/addtt/ladder-vae-pytorch

↓ 7 callersMethodmean
(params)
lib/likelihoods.py:19
↓ 6 callersMethod__init__
(self, channels, merge_type, nonlin=nn.LeakyReLU,
models/lvae_layers.py:329
↓ 3 callersMethod__init__
(self, ch_in, color_channels)
lib/likelihoods.py:83
↓ 3 callersMethod_get_binarized_mnist
Get statically binarized MNIST. Code partially taken from https://github.com/altosaar/proximity_vi/blob/master/get_binary_mnist.py
lib/datasets.py:24
↓ 3 callersFunction_reduce
(x, reduce)
lib/likelihoods.py:414
↓ 3 callersFunctionlog_discretized_logistic
Log of the probability mass of the values x under the logistic distribution with parameters mean and scale. The sum is taken over all dimensi
lib/likelihoods.py:233
↓ 2 callersMethod__init__
(self, channels, kernel_size, nonlin=nn.LeakyReLU)
lib/nn.py:114
↓ 2 callersFunction_input_check
(x, mean, scale_param, reduce)
lib/likelihoods.py:422
↓ 2 callersMethodget_padded_size
Returns the smallest size (H, W) of the image with actual size given as input, such that H and W are powers of 2. :param size
models/lvae.py:327
↓ 2 callersMethodtopdown_pass
(self, bu_values=None, n_img_prior=None, mode_l
models/lvae.py:229
↓ 1 callersMethodbottomup_pass
(self, x)
models/lvae.py:216
↓ 1 callersFunctiondiscretized_mix_logistic_loss
log-likelihood for mixture of discretized logistics, assumes the data has been rescaled to [-1,1] interval Code taken from pytorch adapt
lib/likelihoods.py:291
↓ 1 callersMethoddistr_params
(self, x)
lib/likelihoods.py:15
↓ 1 callersMethodget_top_prior_param_shape
(self, n_imgs=1)
models/lvae.py:364
↓ 1 callersFunctioninspect_layer_repr
(model, img_folder, n=8)
evaluate.py:95
↓ 1 callersFunctionkl_normal_mc
One-sample estimation of element-wise KL between two diagonal multivariate normal distributions. Any number of dimensions, broadcasting s
lib/stochastic.py:209
↓ 1 callersFunctionlog_bernoulli
(x, mean, reduce='mean')
lib/likelihoods.py:385
↓ 1 callersMethodlog_likelihood
(self, x, params)
lib/likelihoods.py:30
↓ 1 callersFunctionlog_normal
Log of the probability density of the values x untder the Normal distribution with parameters mean and logvar. The sum is taken over all
lib/likelihoods.py:391
↓ 1 callersFunctionlogistic_rsample
Returns a sample from Logistic with specified mean and log scale. :param mu_ls: a tensor containing mean and log scale along dim=1,
lib/stochastic.py:115
↓ 1 callersFunctionmain
()
main.py:6
↓ 1 callersFunctionmain
()
evaluate.py:117
↓ 1 callersMethodmode
(params)
lib/likelihoods.py:23
↓ 1 callersMethodpad_input
Pads input x so that its sizes are powers of 2 :param x: :return: Padded tensor
models/lvae.py:317
↓ 1 callersMethodrun
(self)
evaluate.py:20
↓ 1 callersMethodsample
(params)
lib/likelihoods.py:27
↓ 1 callersFunctionsample_from_discretized_mix_logistic
Code taken from pytorch adaptation of original PixelCNN++ tf implementation https://github.com/pclucas14/pixel-cnn-pp
lib/stochastic.py:141
↓ 1 callersMethodsample_prior
(self, n_imgs, mode_layers=None, constant_layers=None)
models/lvae.py:351
↓ 1 callersFunctionto_one_hot
(tensor, n)
lib/stochastic.py:147
Method__init__
(self, channels, nonlin, kernel=None, grou
lib/nn.py:26
Method__init__
(self, *args, **kwargs)
lib/nn.py:104
Method__init__
(self, ch_in, color_channels)
lib/likelihoods.py:53
Method__init__
(self, ch_in, color_channels, n_bins, double=False)
lib/likelihoods.py:131
Method__init__
(self, ch_in, n_components=10)
lib/likelihoods.py:197
Method__init__
(self, folder, train, download=False, shuffle_init=False)
lib/datasets.py:11
Method__init__
(self, c_in, c_vars, c_out, kernel=3, transform_p_params=True)
lib/stochastic.py:15
Method__init__
(self, color_ch, z_dims, blocks_per_layer=2,
models/lvae.py:17
Method__init__
(self, z_dim, n_res_blocks, n_filters, is_
models/lvae_layers.py:29
Method__init__
(self, n_res_blocks, n_filters, downsampling_steps=0,
models/lvae_layers.py:188
Method__init__
(self, mode, c_in, c_out, nonlin=nn.LeakyR
models/lvae_layers.py:241
Method__init__
(self, *args, upsample=False, **kwargs)
models/lvae_layers.py:311
Method__init__
(self, *args, downsample=False, **kwargs)
models/lvae_layers.py:318
Method__init__
(self, channels, nonlin, batchnorm, dropout, res_block_type)
models/lvae_layers.py:370
Method__init__
(self, args, cuda)
experiment/data.py:27
Method_add_args
(self, parser: argparse.ArgumentParser)
evaluate.py:52
Method_add_args
(self, parser: argparse.ArgumentParser)
experiment/experiment_manager.py:107
Method_check_args
(cls, args: argparse.Namespace)
evaluate.py:83
Method_check_args
(cls, args: argparse.Namespace)
experiment/experiment_manager.py:262
Method_define_args_defaults
(cls)
experiment/experiment_manager.py:84
Method_make_datamanager
(self)
experiment/experiment_manager.py:34
Method_make_model
(self)
experiment/experiment_manager.py:38
Method_make_optimizer
(self)
experiment/experiment_manager.py:76
Method_make_run_description
(args: argparse.Namespace)
experiment/experiment_manager.py:293
Function_pad_tensor
(x, size, value=None)
lib/datasets.py:75
Methoddistr_params
(self, x)
lib/likelihoods.py:60
Methoddistr_params
(self, x)
lib/likelihoods.py:90
Methoddistr_params
(self, x)
lib/likelihoods.py:140
Methoddistr_params
(self, x)
lib/likelihoods.py:204
Methodforward
(self, x)
lib/nn.py:98
Methodforward
(self, x)
lib/nn.py:121
Methodforward
(self, input_, x)
lib/likelihoods.py:33
Methodforward
(self, p_params, q_params=None, forced_latent=None,
lib/stochastic.py:29
Methodforward
(self, x)
models/lvae.py:172
Methodforward
(self, input_=None, skip_connection_input=None, inference_mode
models/lvae_layers.py:115
Methodforward
(self, x)
models/lvae_layers.py:218
Methodforward
(self, x)
models/lvae_layers.py:300
Methodforward
(self, x, y)
models/lvae_layers.py:358
Methodforward_pass
(self, x: torch.Tensor, y: Optional[torch.Tensor] = None)
experiment/experiment_manager.py:322
Methodget_metrics_dict
(cls, results: dict)
experiment/experiment_manager.py:403
Methodlist_options
(lst)
experiment/experiment_manager.py:111
Methodlog_likelihood
(self, x, params)
lib/likelihoods.py:77
Methodlog_likelihood
(self, x, params)
lib/likelihoods.py:112
Methodlog_likelihood
(self, x, params)
lib/likelihoods.py:167
Methodlog_likelihood
(self, x, params)
lib/likelihoods.py:227
Methodmean
(params)
lib/likelihoods.py:66
Methodmean
(params)
lib/likelihoods.py:100
Methodmean
(params)
lib/likelihoods.py:153
Methodmean
(params)
lib/likelihoods.py:213
Methodmode
(params)
lib/likelihoods.py:70
Methodmode
(params)
lib/likelihoods.py:104
Methodmode
(params)
lib/likelihoods.py:157
Methodmode
(params)
lib/likelihoods.py:217
Methodsample
(params)
lib/likelihoods.py:74
Methodsample
(params)
lib/likelihoods.py:108
Methodsample
(params)
lib/likelihoods.py:161
Methodsample
(params)
lib/likelihoods.py:221
Methodtest_log_str
(cls, summaries: dict, step: int, epoch: Option
experiment/experiment_manager.py:380
Methodtrain_log_str
(cls, summaries: dict, step: int, epoch: Opt
experiment/experiment_manager.py:370