MCPcopy Create free account

hub / github.com/NVlabs/GSPN / functions

Functions643 in github.com/NVlabs/GSPN

Method__init__
(self, indices)
classification/data/samplers.py:18
Method__init__
(self)
classification/data/zipreader.py:27
Method__init__
(self, *args, **kwargs)
classification/data/build.py:116
Method__init__
(self, root, ann_file='', transform=None, target_transform=None)
classification/data/imagenet22k_dataset.py:19
Method__init__
(self, root, ann_file='', img_prefix='', transform=None, target_transform=None, loader=defaul
classification/data/cached_image_folder.py:228
Method__init__
(self, input_size=192, mask_patch_size=32, model_patch_size=4, mask_ratio=0.6)
classification/data/data_simmim_pt.py:22
Method__init__
(self, config)
classification/data/data_simmim_pt.py:49
Method__init__
(self, dim, out_dim=-1, norm_layer=nn.LayerNorm, channel_first=False)
classification/models/modules.py:51
Method__init__
(self, *args)
classification/models/modules.py:92
Method__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.,channels_first=False)
classification/models/modules.py:122
Method__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.,channels_first=False)
classification/models/modules.py:143
Method__init__
(self, hidden_dim, ffn_dim)
classification/models/modules.py:165
Method__init__
(self, dim)
classification/models/modules.py:180
Method__init__
( self, dim: int, init_values: float = 1e-5, inplace: bool = F
classification/models/modules.py:192
Method__init__
(self, hidden_dim, ffn_dim)
classification/models/modules.py:207
Method__init__
( self, feat_size, items_each_chunk=8, hidden_dim: int = 0, drop_path:
classification/models/gspn.py:213
Method__init__
( self, feat_sizes, items_each_chunk=8, patch_size=4, in_chans=3,
classification/models/gspn.py:300
Method__iter__
(self)
generation/evaluator.py:471
Method__iter__
(self)
generation/extract_features.py:113
Method__iter__
(self)
t2i/src/eval/misc.py:124
Method__iter__
(self)
classification/data/samplers.py:22
Method__len__
(self)
generation/train.py:114
Method__len__
(self)
generation/evaluator.py:468
Method__len__
(self)
generation/extract_features.py:116
Method__len__
(self)
generation/tools/fid_score.py:63
Method__len__
(self)
t2i/src/eval/data_util.py:97
Method__len__
(self)
classification/data/samplers.py:25
Method__len__
(self)
classification/data/build.py:119
Method__len__
(self)
classification/data/imagenet22k_dataset.py:60
Method__repr__
(self)
ops/gaterecurrent/gaterecurrent2dnoind.py:126
Method__repr__
(self)
t2i/src/eval/data_util.py:39
Method__repr__
(self)
classification/data/cached_image_folder.py:164
Method__setattr__
(self, name: str, value: Any)
t2i/src/eval/dnnlib/util.py:48
Method_basic_init
(module)
generation/gspn.py:390
Function_forward
(*args, **kwargs)
t2i/src/tools.py:365
Method_init_weights
(self, m: nn.Module)
classification/models/gspn.py:432
Method_load_from_state_dict
(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs)
generation/gspn.py:50
Method_load_from_state_dict
(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs)
classification/models/modules.py:37
Method_load_from_state_dict
(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs)
classification/models/gspn.py:587
Method_make_downsample
(dim=96, out_dim=192, norm_layer=nn.LayerNorm, channel_first=False)
classification/models/gspn.py:478
Method_make_downsample_v3
(dim=96, out_dim=192, norm_layer=nn.LayerNorm, channel_first=False)
classification/models/gspn.py:488
Method_make_patch_embed
(in_chans=3, embed_dim=96, patch_size=4, patch_norm=True, norm_layer=nn.LayerNorm, channel_first=False)
classification/models/gspn.py:452
Method_make_patch_embed_v2
(in_chans=3, embed_dim=96, patch_size=4, patch_norm=True, norm_layer=nn.LayerNorm, channel_first=False)
classification/models/gspn.py:461
Method_patch_merging_pad_channel_first
(x: torch.Tensor)
classification/models/modules.py:72
Method_patch_merging_pad_channel_last
(x: torch.Tensor)
classification/models/modules.py:60
Function_reconstruct_persistent_obj
r"""Hook that is called internally by the `pickle` module to unpickle a persistent object.
t2i/src/eval/persistence.py:185
Method_scale_timesteps
(self, t)
generation/diffusion/respace.py:118
Methodadd_embedding
(self)
t2i/src/distrifuser/models/distri_sdxl_unet_tp.py:203
Methodadd_embedding
(self)
t2i/src/distrifuser/models/naive_patch_sdxl.py:219
Methodadd_embedding
(self)
t2i/src/distrifuser/models/distri_sdxl_unet_pp.py:218
Methodas_dict
r"""Returns the averages accumulated between the last two calls to `update()` as an `dnnlib.EasyDict`. The contents are as follows:
t2i/src/eval/training_stats.py:211
Functionask_yes_no
Ask the user the question until the user inputs a valid answer.
t2i/src/eval/dnnlib/util.py:166
Functionassert_shape
(tensor, ref_shape)
t2i/src/eval/misc.py:81
Methodbackward
(ctx, grad_output)
ops/gaterecurrent/gaterecurrent2dnoind.py:44
Methodbackward
(ctx, grad_output)
t2i/src/eval/ops/grid_sample_gradfix.py:54
Methodbackward
(ctx, grad2_grad_input, grad2_grad_grid)
t2i/src/eval/ops/grid_sample_gradfix.py:70
Methodbackward
(ctx, d_dx)
t2i/src/eval/ops/bias_act.py:192
Methodbackward
(ctx, dy)
t2i/src/eval/ops/upfirdn2d.py:249
Methodbackward
(ctx, dout)
t2i/src/eval/ops/fma.py:29
Methodbackward
(ctx, grad_output)
t2i/src/eval/ops/conv2d_gradfix.py:119
Methodbackward
(ctx, grad2_grad_weight)
t2i/src/eval/ops/conv2d_gradfix.py:151
Functionbias_act
r"""Fused bias and activation function. Adds bias `b` to activation tensor `x`, evaluates activation function `act`, and scales the result by
t2i/src/eval/ops/bias_act.py:58
Functionbias_act
t2i/src/eval/ops/bias_act.cpp:32
Functionbuild_loader
(config, simmim=False, is_pretrain=False)
classification/data/__init__.py:6
Functionbuild_optimizer_swimmim
Build optimizer, set weight decay of normalization to 0 by default.
classification/utils/optimizer.py:72
Methodcalc_bpd_loop
Compute the entire variational lower-bound, measured in bits-per-dim, as well as other related quantities. :param model: the
generation/diffusion/gaussian_diffusion.py:812
Functioncalculate_fid_given_paths
Calculates the FID of two paths
generation/tools/fid_score.py:239
Methodchange_name
(src, dst, state_dict: dict = state_dict, strict=False)
classification/models/gspn.py:600
Functioncheck_ddp_consistency
(module, ignore_regex=None)
t2i/src/eval/misc.py:180
Methodcheck_name
(src, state_dict: dict = state_dict, strict=False)
classification/models/gspn.py:589
Methodckpt_forward
(*inputs)
generation/gspn.py:421
Functioncleanup
End DDP training.
generation/extract_features.py:62
Methodclip_skip
(self)
t2i/src/pipelines/pipeline_highres_sdxl.py:864
Methodclip_skip
(self)
t2i/src/pipelines/pipeline_superres_sdxl.py:865
Functioncollate_fn
(batch)
classification/data/data_simmim_pt.py:77
Methodcondition_mean
(self, cond_fn, *args, **kwargs)
generation/diffusion/respace.py:105
Methodcondition_score
(self, cond_fn, *args, **kwargs)
generation/diffusion/respace.py:108
Methodconfig
(self)
t2i/src/distrifuser/models/base_model.py:44
Functionconstant
(value, shape=None, dtype=None, device=None, memory_format=None)
t2i/src/eval/misc.py:21
Functionconstruct_class_by_name
Finds the python class with the given name and constructs it with the given arguments.
t2i/src/eval/dnnlib/util.py:300
Functioncontinuous_gaussian_log_likelihood
Compute the log-likelihood of a continuous Gaussian distribution. :param x: the targets :param means: the Gaussian mean Tensor. :para
generation/diffusion/diffusion_utils.py:53
Functionconv2d
(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)
t2i/src/eval/ops/conv2d_gradfix.py:35
Functionconv2d_resample
r"""2D convolution with optional up/downsampling. Padding is performed only once at the beginning, not between the operations. Args:
t2i/src/eval/ops/conv2d_resample.py:59
Functionconv_transpose2d
(input, weight, bias=None, stride=1, padding=0, output_padding=0, groups=1, dilation=1)
t2i/src/eval/ops/conv2d_gradfix.py:40
Functioncopy_files_and_create_dirs
Takes in a list of tuples of (src, dst) paths and copies files. Will create all necessary directories.
t2i/src/eval/dnnlib/util.py:361
Functioncopy_params_and_buffers
(src_module, dst_module, require_all=False)
t2i/src/eval/misc.py:155
Functioncreate_logger
Create a logger that writes to a log file and stdout.
generation/extract_features.py:69
Functioncreate_named_schedule_sampler
Create a ScheduleSampler from a library of pre-defined samplers. :param name: the name of the sampler. :param diffusion: the diffusion ob
generation/diffusion/timestep_sampler.py:19
Methodcreate_padded_x
()
t2i/src/distrifuser/modules/pp/conv2d.py:72
Methodcross_attention_kwargs
(self)
t2i/src/pipelines/pipeline_highres_sdxl.py:875
Methodcross_attention_kwargs
(self)
t2i/src/pipelines/pipeline_superres_sdxl.py:876
Functioncustom_forward
(*inputs)
t2i/src/tools.py:510
Methodddim_reverse_sample
Sample x_{t+1} from the model using DDIM reverse ODE.
generation/diffusion/gaussian_diffusion.py:568
Methodddim_sample_loop
Generate samples from the model using DDIM. Same usage as p_sample_loop().
generation/diffusion/gaussian_diffusion.py:606
Functionddp_sync
(module, sync)
t2i/src/eval/misc.py:169
Functiondecorator
(*args, **kwargs)
t2i/src/eval/misc.py:100
Functiondefault_img_loader
(path)
classification/data/cached_image_folder.py:201
Methoddo_classifier_free_guidance
(self)
t2i/src/pipelines/pipeline_highres_sdxl.py:871
Methoddo_classifier_free_guidance
(self)
t2i/src/pipelines/pipeline_superres_sdxl.py:872
Functiondownsample2d
r"""Downsample a batch of 2D images using the given 2D FIR filter. By default, the result is padded so that its shape is a fraction of the input.
t2i/src/eval/ops/upfirdn2d.py:350
← previousnext →401–500 of 643, ranked by callers