Functions — github.com/baofff/Extended-Analytic-DPM

hub / github.com/baofff/Extended-Analytic-DPM / functions

Functions635 in github.com/baofff/Extended-Analytic-DPM

⨍ Functions 635◇ Types & classes 128

Method__init__

(self, schedule, ratio, wrapper, **kwargs)

core/criterions/ddpm.py:71

Method__init__

(self, schedule, ratio, wrapper, **kwargs)

core/criterions/ddpm.py:88

Method__init__

(self, sde, wrapper, **kwargs)

core/criterions/ddpm.py:105

Method__init__

(self, sde, ratio, wrapper, **kwargs)

core/criterions/ddpm.py:116

Method__init__

(self, sde, ratio, wrapper, **kwargs)

core/criterions/ddpm.py:133

Method__init__

(self, schedule_name, N)

core/diffusion/schedule.py:64

Method__init__

(self, wrapper, schedule, clip_x0: bool, clip_cov_x0=None, avg_cov=False)

core/diffusion/dtdpm.py:14

Method__init__

(self, wrapper, schedule, clip_x0: bool, eta: float, clip_cov_x0=None, avg_cov=False)

core/diffusion/dtdpm.py:188

Method__init__

(self, sde, wrapper)

core/diffusion/sde.py:38

Method__init__

(self, sde, wrapper)

core/diffusion/sde.py:65

Method__init__

(self, beta_min=0.1, beta_max=20)

core/diffusion/sde.py:88

Method__init__

(self, beta_min=0.1, beta_max=20)

core/diffusion/sde.py:140

Method__init__

(self, model0: nn.Module, model1: nn.Module, split_idx: int, typ: str, rescale_timesteps: bool,

core/diffusion/wrapper.py:51

Method__init__

(self, model: nn.Module, typ: str, N, T=1, bipartition=None)

core/diffusion/wrapper.py:78

Method__init__

(self, model: nn.Module, typ: str, T=1, bipartition=None)

core/diffusion/wrapper.py:103

Method__init__

r""" Manage a dict of models (nn.Modules)

core/utils/managers.py:79

Method__init__

r""" Manage a dict of optimizers (optim.Optimizer)

core/utils/managers.py:144

Method__init__

r""" Manage a dict of lr_schedulers (optim.lr_scheduler._LRScheduler)

core/utils/managers.py:152

Method__init__

(self, model: [nn.Module, None])

core/utils/compatible.py:6

Method__init__

(self, *objs: Union[torch.Tensor, nn.Module], requires_grad: Union[List[bool], bool])

core/func/differential.py:21

Method__init__

(self, head_out_ch, mode="simple", **kwargs)

libs/ddpm/customized.py:7

Method__init__

(self, in_channels, with_conv)

libs/ddpm/model.py:37

Method__init__

(self, in_channels, with_conv)

libs/ddpm/model.py:55

Method__init__

(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout, temb_channels=512)

libs/ddpm/model.py:77

Method__init__

(self, in_channels)

libs/ddpm/model.py:137

Method__init__

(self, channels, use_conv, dims=2)

libs/iddpm/unet.py:91

Method__init__

( self, channels, emb_channels, dropout, out_channels=None, us

libs/iddpm/unet.py:122

Method__init__

(self, channels, num_heads=1, use_checkpoint=False)

libs/iddpm/unet.py:208

Method__init__

( self, in_channels, model_channels, out_channels, num_res_blocks,

libs/iddpm/unet.py:301

Method__init__

(self, in_channels, *args, **kwargs)

libs/iddpm/unet.py:536

Method__init__

(self, **kwargs)

libs/iddpm/customized.py:15

Method__init__

(self, config)

libs/score_sde/customized.py:23

Method__init__

(self, channel, negative_slope=0.2, scale=2 ** 0.5)

libs/score_sde/op/fused_act.py:75

Method__init__

(self)

libs/score_sde/models/layers.py:96

Method__init__

(self, features, n_stages, act=nn.ReLU(), maxpool=True)

libs/score_sde/models/layers.py:134

Method__init__

(self, features, n_stages, num_classes, normalizer, act=nn.ReLU())

libs/score_sde/models/layers.py:158

Method__init__

(self, features, n_blocks, n_stages, act=nn.ReLU())

libs/score_sde/models/layers.py:184

Method__init__

(self, features, n_blocks, n_stages, num_classes, normalizer, act=nn.ReLU())

libs/score_sde/models/layers.py:208

Method__init__

(self, in_planes, features)

libs/score_sde/models/layers.py:235

Method__init__

(self, in_planes, features, num_classes, normalizer)

libs/score_sde/models/layers.py:254

Method__init__

(self, in_planes, features, act=nn.ReLU(), start=False, end=False, maxpool=True)

libs/score_sde/models/layers.py:278

Method__init__

(self, in_planes, features, num_classes, normalizer, act=nn.ReLU(), start=False, end=False)

libs/score_sde/models/layers.py:314

Method__init__

(self, input_dim, output_dim, kernel_size=3, biases=True, adjust_padding=False)

libs/score_sde/models/layers.py:352

Method__init__

(self, input_dim, output_dim, kernel_size=3, biases=True)

libs/score_sde/models/layers.py:373

Method__init__

(self, input_dim, output_dim, kernel_size=3, biases=True)

libs/score_sde/models/layers.py:385

Method__init__

(self, input_dim, output_dim, num_classes, resample=1, act=nn.ELU(), normalization=ConditionalI

libs/score_sde/models/layers.py:398

Method__init__

(self, input_dim, output_dim, resample=None, act=nn.ELU(), normalization=nn.InstanceNorm2d, adj

libs/score_sde/models/layers.py:454

Method__init__

(self, channels)

libs/score_sde/models/layers.py:560

Method__init__

(self, channels, with_conv=False)

libs/score_sde/models/layers.py:585

Method__init__

(self, channels, with_conv=False)

libs/score_sde/models/layers.py:600

Method__init__

(self, act, in_ch, out_ch=None, temb_dim=None, conv_shortcut=False, dropout=0.1)

libs/score_sde/models/layers.py:621

Method__init__

Args: parameters: Iterable of `torch.nn.Parameter`; usually the result of `model.parameters()`. decay: The exponential decay.

libs/score_sde/models/ema.py:15

Method__init__

(self, config)

libs/score_sde/models/ddpm.py:41

Method__init__

(self, config)

libs/score_sde/models/ncsnpp.py:38

Method__init__

(self, embedding_size=256, scale=1.0)

libs/score_sde/models/layerspp.py:35

Method__init__

(self, channels, skip_rescale=False, init_scale=0.)

libs/score_sde/models/layerspp.py:65

Method__init__

(self, in_ch=None, out_ch=None, with_conv=False, fir=False, fir_kernel=(1, 3, 3, 1))

libs/score_sde/models/layerspp.py:95

Method__init__

(self, in_ch=None, out_ch=None, with_conv=False, fir=False, fir_kernel=(1, 3, 3, 1))

libs/score_sde/models/layerspp.py:130

Method__init__

(self, act, in_ch, out_ch=None, temb_dim=None, conv_shortcut=False, dropout=0.1, skip_rescale=F

libs/score_sde/models/layerspp.py:169

Method__init__

(self, act, in_ch, out_ch=None, temb_dim=None, up=False, down=False, dropout=0.1, fir=False, fi

libs/score_sde/models/layerspp.py:213

Method__init__

(self, config)

libs/score_sde/models/ncsnv2.py:45

Method__init__

(self, config)

libs/score_sde/models/ncsnv2.py:224

Method__init__

(self, config)

libs/score_sde/models/ncsnv2.py:318

Method__init__

(self, num_features, num_classes, bias=True)

libs/score_sde/models/normalization.py:44

Method__init__

(self, num_features, num_classes, bias=True)

libs/score_sde/models/normalization.py:69

Method__init__

(self, num_features, num_classes, bias=False)

libs/score_sde/models/normalization.py:94

Method__init__

(self, num_features, bias=False)

libs/score_sde/models/normalization.py:111

Method__init__

(self, num_features, num_classes, bias=True)

libs/score_sde/models/normalization.py:127

Method__init__

(self, num_features, bias=True)

libs/score_sde/models/normalization.py:158

Method__init__

(self, num_features, num_classes, bias=True)

libs/score_sde/models/normalization.py:187

Method__init__

(self, in_ch, out_ch, kernel, up=False, down=False, resample_kernel=(1, 3, 3, 1),

libs/score_sde/models/up_or_down_sampling.py:26

Method__len__

(self)

tools/fid_score.py:63

Method__len__

(self)

interface/datasets/utils.py:49

Method__len__

(self)

interface/datasets/utils.py:65

Method__len__

(self)

interface/datasets/utils.py:81

Method__len__

(self)

interface/datasets/utils.py:93

Method__len__

(self)

interface/datasets/celeba.py:299

Method__len__

(self)

interface/datasets/imagenet64.py:15

Method__len__

(self)

interface/datasets/lsun/lsun.py:88

Method__len__

(self)

interface/datasets/lsun/lsun.py:202

Method__repr__

(self)

interface/datasets/utils.py:39

Method__repr__

(self)

interface/datasets/celeba.py:114

Method__repr__

(self)

interface/datasets/lsun/vision.py:75

Method__repr__

(self)

core/diffusion/schedule.py:59

Method__repr__

(self)

core/diffusion/schedule.py:70

Method_format_transform_repr

(self, transform: Callable, head: str)

interface/datasets/celeba.py:88

Method_forward

(self, x, emb)

libs/iddpm/unet.py:184

Method_forward

(self, x)

libs/iddpm/unet.py:222

Method_predict_cov_prev

(self, s, t, typ, cov_x0_pred=None)

core/diffusion/dtdpm.py:203

Function_solve_fn_dp_general

(fn, a, b, K)

core/diffusion/trajectory.py:114

Methodaffine_transform

(self, dataset)

interface/datasets/lsun_bedroom.py:17

Methodaffine_transform

(self, dataset)

interface/datasets/celeba.py:358

Methodaffine_transform

(self, dataset)

interface/datasets/cifar10.py:33

Methodaffine_transform

(self, dataset)

interface/datasets/imagenet64.py:40

Methodaffine_transform

(self, dataset)

interface/datasets/mnist.py:54

Methodallow_labelled

(self)

interface/datasets/dataset_factory.py:18

Methodbackward

(ctx, gradgrad_input, gradgrad_bias)

libs/score_sde/op/fused_act.py:43

Methodbackward

(ctx, grad_output)

libs/score_sde/op/fused_act.py:64

Methodbackward

(ctx, gradgrad_input)

libs/score_sde/op/upfirdn2d.py:63

Methodbackward

(ctx, grad_output)

libs/score_sde/op/upfirdn2d.py:127

← previous next →301–400 of 635, ranked by callers