MCPcopy Create free account

hub / github.com/NVlabs/GSPN / functions

Functions643 in github.com/NVlabs/GSPN

↓ 1 callersMethodspn
(self, x)
generation/gspn.py:222
↓ 1 callersMethodspn_block
(self, X, l, u, Gl, Gm, Gr, D=None, spn_module=None)
generation/gspn.py:212
↓ 1 callersMethodspn_block
(self, X, l, u, Gl, Gm, Gr, D=None, spn_module=None)
classification/models/gspn.py:123
↓ 1 callersFunctiontensor2pil
output image : tensor to PIL
t2i/src/eval/calculate_metrics.py:22
↓ 1 callersMethodtimestep_embedding
Create sinusoidal timestep embeddings. :param t: a 1-D Tensor of N indices, one per batch element. These ma
generation/gspn.py:100
↓ 1 callersMethodtoken_drop
Drops labels to enable classifier-free guidance.
generation/gspn.py:137
↓ 1 callersFunctiontrain_one_epoch
(config, model, criterion, data_loader, optimizer, epoch, mixup_fn, lr_scheduler, loss_scaler, model_ema=None,
classification/main.py:208
↓ 1 callersMethodtraining_losses
( self, model, *args, **kwargs )
generation/diffusion/respace.py:100
↓ 1 callersFunctionunpatchify
(x, channels=3)
generation/gspn.py:37
↓ 1 callersMethodupcast_vae
(self)
t2i/src/pipelines/pipeline_superres_sdxl.py:807
↓ 1 callersFunctionupdate_config
(config, args)
classification/config.py:234
↓ 1 callersMethodupdate_with_all_losses
Update the reweighting using losses from a model. Sub-classes should override this method to update the reweighting using los
generation/diffusion/timestep_sampler.py:112
↓ 1 callersMethodwarmup
(self)
generation/evaluator.py:143
↓ 1 callersMethodweights
Get a numpy array of weights, one per diffusion step. The weights needn't be normalized, but must be positive.
generation/diffusion/timestep_sampler.py:44
FunctionGSPN_B_2
(**kwargs)
generation/gspn.py:580
FunctionGSPN_B_4
(**kwargs)
generation/gspn.py:583
FunctionGSPN_L_2
(**kwargs)
generation/gspn.py:586
FunctionGSPN_L_4
(**kwargs)
generation/gspn.py:589
FunctionGSPN_S_2
(**kwargs)
generation/gspn.py:574
FunctionGSPN_S_4
(**kwargs)
generation/gspn.py:577
FunctionGSPN_XL_2
(**kwargs)
generation/gspn.py:592
FunctionGSPN_XL_4
(**kwargs)
generation/gspn.py:595
FunctionPYBIND11_MODULE
ops/gaterecurrent/src/gaterecurrent2dnoind_cuda.cpp:110
FunctionPYBIND11_MODULE
t2i/src/eval/ops/bias_act.cpp:94
FunctionPYBIND11_MODULE
t2i/src/eval/ops/upfirdn2d.cpp:98
Method__call__
(self, loss, optimizer, clip_grad=None, parameters=None, create_graph=False, update_grad=True)
generation/train.py:134
Method__call__
(self, x, ts, **kwargs)
generation/diffusion/respace.py:130
Method__call__
(self, img)
t2i/src/eval/data_util.py:36
Method__call__
r""" Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*):
t2i/src/pipelines/pipeline_highres_sdxl.py:888
Method__call__
r""" Function invoked when calling the pipeline for generation. Args: image: The image required for super
t2i/src/pipelines/pipeline_superres_sdxl.py:889
Method__call__
(self, *args, **kwargs)
t2i/src/pipelines/pipeline_distrifusion_sdxl.py:59
Method__call__
(self, loss, optimizer, clip_grad=None, parameters=None, create_graph=False, update_grad=True)
classification/utils/utils.py:172
Method__call__
(self)
classification/data/data_simmim_pt.py:37
Method__call__
(self, img)
classification/data/data_simmim_pt.py:70
Method__delattr__
(self, name: str)
t2i/src/eval/dnnlib/util.py:51
Method__enter__
(self)
t2i/src/eval/dnnlib/util.py:71
Method__exit__
(self, exc_type: Any, exc_value: Any, traceback: Any)
t2i/src/eval/dnnlib/util.py:74
Method__getattr__
(self, name: str)
t2i/src/eval/dnnlib/util.py:42
Method__getitem__
(self, idx)
generation/train.py:119
Method__getitem__
(self, i)
generation/tools/fid_score.py:66
Method__getitem__
(self, index)
t2i/src/eval/data_util.py:101
Method__getitem__
r"""Convenience getter. `collector[name]` is a synonym for `collector.mean(name)`.
t2i/src/eval/training_stats.py:225
Method__getitem__
(self, *args,**kwargs)
classification/data/build.py:122
Method__getitem__
Args: index (int): Index Returns: tuple: (image, target) where target is class_index of the target class.
classification/data/imagenet22k_dataset.py:39
Method__getitem__
Args: index (int): Index Returns: tuple: (sample, target) where target is class_index of the target class.
classification/data/cached_image_folder.py:145
Method__getitem__
Args: index (int): Index Returns: tuple: (image, target) where target is class_index of the target class.
classification/data/cached_image_folder.py:236
Method__init__
(self, items_each_chunk_, backend='cuda')
ops/gaterecurrent/gaterecurrent2dnoind.py:114
Method__init__
(self, features_dir, labels_dir)
generation/train.py:107
Method__init__
(self)
generation/train.py:131
Method__init__
(self, mu: np.ndarray, sigma: np.ndarray)
generation/evaluator.py:76
Method__init__
( self, session, batch_size=64, softmax_batch_size=512, )
generation/evaluator.py:127
Method__init__
Estimate the manifold of given feature vectors. :param session: the TensorFlow session. :param row_batch_size: row batch siz
generation/evaluator.py:220
Method__init__
(self, session)
generation/evaluator.py:377
Method__init__
(self, gen_fn, length)
generation/evaluator.py:464
Method__init__
(self, arr_f, shape, dtype)
generation/evaluator.py:476
Method__init__
(self, arr)
generation/evaluator.py:502
Method__init__
(self, data_source, start_index=0)
generation/extract_features.py:109
Method__init__
(self, hidden_size, patch_size, out_channels, condition=False)
generation/gspn.py:59
Method__init__
(self, hidden_size, frequency_embedding_size=256)
generation/gspn.py:90
Method__init__
(self, num_classes, hidden_size, dropout_prob)
generation/gspn.py:130
Method__init__
( self, input_size, hidden_size, items_each_chunk=2, norm_cls=nn.LayerNorm, drop_path=0.,
generation/gspn.py:157
Method__init__
(self, files, transforms=None)
generation/tools/fid_score.py:59
Method__init__
(self, in_channels, pool_features)
generation/tools/inception.py:226
Method__init__
(self, in_channels, channels_7x7)
generation/tools/inception.py:251
Method__init__
(self, in_channels)
generation/tools/inception.py:279
Method__init__
(self, in_channels)
generation/tools/inception.py:312
Method__init__
(self, diffusion)
generation/diffusion/timestep_sampler.py:69
Method__init__
(self, diffusion, history_per_term=10, uniform_prob=0.001)
generation/diffusion/timestep_sampler.py:127
Method__init__
( self, *, betas, model_mean_type, model_var_type, loss_type
generation/diffusion/gaussian_diffusion.py:159
Method__init__
(self, use_timesteps, **kwargs)
generation/diffusion/respace.py:79
Method__init__
Args: query_dim: the dimension of the query. out_dim: the dimension of the output. dim_head: the dimensio
t2i/src/fusion/gspn.py:38
Method__init__
(self, modules_list, *args, **kwargs)
t2i/src/fusion/gspnfusion.py:30
Method__init__
( self, height: int = 1024, width: int = 1024, do_classifier_free_guidance: bo
t2i/src/distrifuser/utils.py:25
Method__init__
(self, distri_config: DistriConfig)
t2i/src/distrifuser/utils.py:114
Method__init__
( self, module: nn.Module, distri_config: DistriConfig, )
t2i/src/distrifuser/modules/base_module.py:7
Method__init__
(self, module: Attention, distri_config: DistriConfig)
t2i/src/distrifuser/modules/pp/attn.py:42
Method__init__
(self, module: Attention, distri_config: DistriConfig)
t2i/src/distrifuser/modules/pp/attn.py:106
Method__init__
(self, module: Attention, distri_config: DistriConfig)
t2i/src/distrifuser/modules/pp/attn.py:196
Method__init__
(self, module: nn.GroupNorm, distri_config: DistriConfig)
t2i/src/distrifuser/modules/pp/groupnorm.py:10
Method__init__
(self, module: nn.Conv2d, distri_config: DistriConfig, is_first_layer: bool = False)
t2i/src/distrifuser/modules/pp/conv2d.py:11
Method__init__
(self, module: Attention, distri_config: DistriConfig)
t2i/src/distrifuser/modules/tp/attention.py:12
Method__init__
(self, module: nn.Conv2d, distri_config: DistriConfig)
t2i/src/distrifuser/modules/tp/conv2d.py:11
Method__init__
(self, module: FeedForward, distri_config: DistriConfig)
t2i/src/distrifuser/modules/tp/feed_forward.py:12
Method__init__
(self, module: ResnetBlock2D, distri_config: DistriConfig)
t2i/src/distrifuser/modules/tp/resnet.py:12
Method__init__
(self, model: UNet2DConditionModel, distri_config: DistriConfig)
t2i/src/distrifuser/models/distri_sdxl_unet_tp.py:18
Method__init__
(self, model: nn.Module, distri_config: DistriConfig)
t2i/src/distrifuser/models/base_model.py:9
Method__init__
(self, model: UNet2DConditionModel, distri_config: DistriConfig)
t2i/src/distrifuser/models/naive_patch_sdxl.py:12
Method__init__
(self, model: UNet2DConditionModel, distri_config: DistriConfig)
t2i/src/distrifuser/models/distri_sdxl_unet_pp.py:18
Method__init__
(self, device, seeds)
t2i/src/eval/eval.py:35
Method__init__
(self, dataset, rank=0, num_replicas=1, shuffle=True, seed=0, window_size=0.5)
t2i/src/eval/misc.py:111
Method__init__
(self, data_name, data_dir, captionfile, c
t2i/src/eval/data_util.py:44
Method__init__
(self, *args, **kwargs)
t2i/src/eval/persistence.py:105
Method__init__
(self, regex='.*', keep_previous=True)
t2i/src/eval/training_stats.py:132
Method__init__
(self, file_name: Optional[str] = None, file_mode: str = "w", should_flush: bool = True)
t2i/src/eval/dnnlib/util.py:58
Method__init__
( self, vae: AutoencoderKL, text_encoder: CLIPTextModel, text_encoder_2: CLIPT
t2i/src/pipelines/pipeline_highres_sdxl.py:270
Method__init__
( self, vae: AutoencoderKL, text_encoder: CLIPTextModel, text_encoder_2: CLIPT
t2i/src/pipelines/pipeline_superres_sdxl.py:271
Method__init__
(self, pipeline: StableDiffusionXLPipeline, module_config: DistriConfig)
t2i/src/pipelines/pipeline_distrifusion_sdxl.py:26
Method__init__
(self)
classification/utils/utils.py:169
Method__init__
(self, optimizer: torch.optim.Optimizer, t_initial: int, t_
classification/utils/cosine_lr.py:30
Method__init__
(self, optimizer: torch.optim.Optimizer, milestones, gamma=0.1, warmup_t=0, warmup_lr_init=0, t_in_epochs=True
classification/utils/lr_scheduler.py:123
← previousnext →301–400 of 643, ranked by callers