Functions — github.com/bahjat-kawar/ddrm

CodeHub MCPcopy Create free account

hub / github.com/bahjat-kawar/ddrm / functions

Functions294 in github.com/bahjat-kawar/ddrm

⨍ Functions 294◇ Types & classes 47

↓ 1 callersFunctionparse_args_and_config

↓ 1 callersFunctionprofile_kv

guided_diffusion/logger.py:294

↓ 1 callersMethodsample

runners/diffusion.py:93

↓ 1 callersMethodsample_image

(self, x, model, H_funcs, y_0, sigma_0, last=True, cls_fn=None, classes=None)

runners/diffusion.py:333

↓ 1 callersMethodsample_sequence

(self, model, cls_fn=None)

runners/diffusion.py:165

↓ 1 callersMethodset_comm

guided_diffusion/logger.py:385

↓ 1 callersMethodset_level

guided_diffusion/logger.py:382

↓ 1 callersFunctionsigmoid

runners/diffusion.py:23

↓ 1 callersMethodsingulars

functions/svd_replacement.py:387

↓ 1 callersFunctionsr_create_model

( large_size, small_size, num_channels, num_res_blocks, learn_sigma, class_cond, u

guided_diffusion/script_util.py:335

↓ 1 callersFunctionstate_dict_to_master_params

(model, state_dict, use_fp16)

guided_diffusion/fp16_util.py:116

↓ 1 callersMethodwritekvs

guided_diffusion/logger.py:27

↓ 1 callersMethodwriteseq

guided_diffusion/logger.py:32

↓ 1 callersFunctionzero_grad

guided_diffusion/fp16_util.py:133

Multiplies the input vector by H transposed

functions/svd_replacement.py:54

functions/svd_replacement.py:95

functions/svd_replacement.py:132

functions/svd_replacement.py:158

functions/svd_replacement.py:212

functions/svd_replacement.py:255

functions/svd_replacement.py:299

functions/svd_replacement.py:369

functions/svd_replacement.py:442

functions/svd_replacement.py:520

functions/svd_replacement.py:98

functions/svd_replacement.py:135

functions/svd_replacement.py:161

functions/svd_replacement.py:215

functions/svd_replacement.py:258

functions/svd_replacement.py:302

functions/svd_replacement.py:452

functions/svd_replacement.py:530

functions/svd_replacement.py:89

functions/svd_replacement.py:118

functions/svd_replacement.py:152

functions/svd_replacement.py:182

functions/svd_replacement.py:237

functions/svd_replacement.py:291

functions/svd_replacement.py:424

functions/svd_replacement.py:502

functions/svd_replacement.py:92

functions/svd_replacement.py:125

functions/svd_replacement.py:155

functions/svd_replacement.py:197

functions/svd_replacement.py:246

functions/svd_replacement.py:296

functions/svd_replacement.py:434

functions/svd_replacement.py:512

(self, input, target)

datasets/vision.py:63

datasets/__init__.py:21

Args: img (PIL Image): Image to be cropped. Returns: PIL Image: Cropped image.

datasets/imagenet_subset.py:13

Method__getitem__

datasets/vision.py:28

Method__getitem__

datasets/lsun.py:38

Method__getitem__

Args: index (int): Index Returns: tuple: Tuple (image, target) where target is the index of the target categ

datasets/lsun.py:147

Method__getitem__

datasets/celeba.py:133

Method__getitem__

datasets/imagenet_subset.py:93

(self, args, config, device=None)

runners/diffusion.py:56

( self, spacial_dim: int, embed_dim: int, num_heads_channels: int, out

guided_diffusion/unet.py:27

(self, channels, use_conv, dims=2, out_channels=None)

guided_diffusion/unet.py:123

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

guided_diffusion/unet.py:160

( self, channels, num_heads=1, num_head_channels=-1, use_checkpoint=Fa

guided_diffusion/unet.py:267

(self, n_heads)

guided_diffusion/unet.py:333

(self, n_heads)

guided_diffusion/unet.py:366

( self, image_size, in_channels, model_channels, out_channels,

guided_diffusion/unet.py:427

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

guided_diffusion/unet.py:674

( self, image_size, in_channels, model_channels, out_channels,

guided_diffusion/unet.py:691

( self, *, model, use_fp16=False, fp16_scale_growth=1e-3, init

guided_diffusion/fp16_util.py:149

(self, filename_or_file)

guided_diffusion/logger.py:37

(self, filename)

guided_diffusion/logger.py:99

(self, filename)

guided_diffusion/logger.py:114

guided_diffusion/logger.py:155

(self, dir, output_formats, comm=None)

guided_diffusion/logger.py:337

functions/svd_replacement.py:80

(self, channels, img_dim, missing_indices, device)

functions/svd_replacement.py:111

(self, channels, img_dim, device)

functions/svd_replacement.py:149

(self, channels, img_dim, ratio, device)

functions/svd_replacement.py:172

(self, img_dim, device)

functions/svd_replacement.py:229

(self, channels, img_dim, ratio, perm, device)

functions/svd_replacement.py:284

(self, kernel, channels, img_dim, device, stride = 1)

functions/svd_replacement.py:323

(self, kernel, channels, img_dim, device, ZERO = 3e-2)

functions/svd_replacement.py:406

(self, kernel1, kernel2, channels, img_dim, device)

functions/svd_replacement.py:476

(self, root, transforms=None, transform=None, target_transform=None)

datasets/vision.py:9

(self, transform=None, target_transform=None)

datasets/vision.py:59

(self, x1, x2, y1, y2)

datasets/__init__.py:15

(self, root, transform=None, target_transform=None)

datasets/lsun.py:12

(self, root, split="train", target_type="attr", transform=N

datasets/celeba.py:50

(self, root_dir, meta_file, transform=None,

datasets/imagenet_subset.py:50

(self, in_channels, with_conv)

models/diffusion.py:37

(self, in_channels, with_conv)

models/diffusion.py:56

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

models/diffusion.py:78

(self, in_channels)

models/diffusion.py:138

datasets/lsun.py:57

datasets/lsun.py:172

datasets/celeba.py:158

datasets/imagenet_subset.py:90

datasets/vision.py:75

datasets/__init__.py:24

datasets/imagenet_subset.py:22

guided_diffusion/unet.py:236

guided_diffusion/unet.py:299

← previous next →101–200 of 294, ranked by callers