MCPcopy Create free account

hub / github.com/VITA-Group/TransGAN / functions

Functions560 in github.com/VITA-Group/TransGAN

↓ 62 callersFunctiontrunc_normal_
r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi
models_search/ViT_helper.py:93
↓ 35 callersMethodmean
r"""Returns the mean of the scalars that were accumulated for the given statistic between the last two calls to `update()`, or NaN if
torch_utils/training_stats.py:188
↓ 11 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom_scale2.py:77
↓ 11 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/Celeba256_dis.py:79
↓ 11 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_scale3_local_new_rp.py:77
↓ 11 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom_scale2_rp_noise.py:77
↓ 11 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/Celeba256_gen.py:51
↓ 10 callersFunctionDiffAugment
(x, policy='', channels_first=True, affine=None)
models_search/diff_aug.py:48
↓ 10 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom_rp.py:53
↓ 10 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom_local544444_256_rp_noise.py:48
↓ 10 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom_local544444_256_rp.py:48
↓ 10 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom.py:53
↓ 7 callersFunctionconstant
(value, shape=None, dtype=None, device=None, memory_format=None)
models_search/ada.py:6
↓ 7 callersFunctionmatrix
(*rows, device=None)
models_search/ada.py:27
↓ 6 callersFunction_conv2d_wrapper
Wrapper for the underlying `conv2d()` and `conv_transpose2d()` implementations.
torch_utils/ops/conv2d_resample.py:29
↓ 6 callersFunction_parse_padding
(padding)
torch_utils/ops/upfirdn2d.py:46
↓ 6 callersFunction_parse_scaling
(scaling)
torch_utils/ops/upfirdn2d.py:37
↓ 6 callersMethodclose
Flush, close possible files, and remove stdout/stderr mirroring.
dnnlib/util.py:100
↓ 5 callersFunction_get_filter_size
(f)
torch_utils/ops/upfirdn2d.py:57
↓ 5 callersFunctionload_params
(model, new_param, args, mode="gpu")
functions.py:419
↓ 5 callersFunctionscale2d_inv
(sx, sy, **kwargs)
models_search/ada.py:87
↓ 5 callersFunctiontqdm
(x)
utils/torch_fid_score.py:38
↓ 4 callersMethod__init__
Build pretrained InceptionV3 Parameters ---------- output_blocks : list of int Indices of blocks to return feature
utils/inception_model.py:31
↓ 4 callersMethod__init__
Build pretrained InceptionV3 Parameters ---------- output_blocks : list of int Indices of blocks to return feature
utils/inception.py:31
↓ 4 callersFunction_conv2d_gradfix
(transpose, weight_shape, stride, padding, output_padding, dilation, groups)
torch_utils/ops/conv2d_gradfix.py:68
↓ 4 callersFunction_ntuple
(n)
models_search/ViT_helper.py:37
↓ 4 callersFunction_tuple_of_ints
(xs, ndim)
torch_utils/ops/conv2d_gradfix.py:58
↓ 4 callersFunctioncopy_params
(model, mode='cpu')
functions.py:431
↓ 4 callersMethodflush
Flush written text to both stdout and a file, if open.
dnnlib/util.py:93
↓ 4 callersFunctionhas_same_layout
torch_utils/ops/bias_act.cpp:16
↓ 4 callersMethodstep
(self, current_step)
functions.py:408
↓ 4 callersFunctionwindow_partition
Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
models_search/ViT_custom_local544444_256_rp_noise.py:210
↓ 4 callersFunctionwindow_partition
Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
models_search/ViT_custom_local544444_256_rp.py:203
↓ 4 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
models_search/ViT_custom_local544444_256_rp_noise.py:222
↓ 4 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
models_search/ViT_custom_local544444_256_rp.py:215
↓ 3 callersMethod_get_delta
r"""Returns the raw moments that were accumulated for the given statistic between the last two calls to `update()`, or zero if no scal
torch_utils/training_stats.py:170
↓ 3 callersFunction_unbroadcast
(x, shape)
torch_utils/ops/fma.py:49
↓ 3 callersFunctioncompute_gradient_penalty
Calculates the gradient penalty loss for WGAN GP
functions.py:48
↓ 3 callersFunctioncreate_logger
(log_dir, phase='train')
utils/utils.py:152
↓ 3 callersFunctionget_obj_from_module
Traverses the object name and returns the last (rightmost) python object.
dnnlib/util.py:263
↓ 3 callersFunctionnamed_params_and_buffers
(module)
torch_utils/misc.py:149
↓ 3 callersFunctionpixel_upsample
(x, H, W)
models_search/ViT_custom_local544444_256_rp_noise.py:188
↓ 3 callersFunctionpixel_upsample
(x, H, W)
models_search/ViT_custom_local544444_256_rp.py:181
↓ 3 callersFunctionpixel_upsample
(x, H, W)
models_search/Celeba256_gen.py:247
↓ 3 callersFunctionscale2d
(sx, sy, **kwargs)
models_search/ada.py:52
↓ 3 callersFunctiontranslate2d
(tx, ty, **kwargs)
models_search/ada.py:37
↓ 3 callersMethodupdate
r"""Copies current values of the internal counters to the user-visible state and resets them for the next round. If `keep_previous=Tr
torch_utils/training_stats.py:147
↓ 3 callersFunctionupfirdn2d
r"""Pad, upsample, filter, and downsample a batch of 2D images. Performs the following sequence of operations for each channel: 1. Upsample
torch_utils/ops/upfirdn2d.py:120
↓ 3 callersMethodwrite
Write text to stdout (and a file) and optionally flush.
dnnlib/util.py:78
↓ 2 callersFunction_compute_statistics_of_path
(args, path, model, batch_size, dims, cuda)
utils/torch_fid_score.py:237
↓ 2 callersFunction_get_inception_layer
Prepares inception net for batched usage and returns pool_3 layer.
utils/fid_score.py:54
↓ 2 callersFunction_get_weight_shape
(w)
torch_utils/ops/conv2d_resample.py:21
↓ 2 callersFunction_handle_path
(path, sess, low_profile=False)
utils/fid_score.py:295
↓ 2 callersFunction_should_use_custom_op
(input)
torch_utils/ops/conv2d_gradfix.py:47
↓ 2 callersFunction_upfirdn2d_cuda
Fast CUDA implementation of `upfirdn2d()` using custom ops.
torch_utils/ops/upfirdn2d.py:214
↓ 2 callersMethodbackward
(ctx, dy)
torch_utils/ops/bias_act.py:161
↓ 2 callersFunctionbicubic_upsample
(x, H, W)
models_search/ViT_custom_local544444_256_rp_noise.py:199
↓ 2 callersFunctionbicubic_upsample
(x, H, W)
models_search/ViT_custom_local544444_256_rp.py:192
↓ 2 callersFunctionbicubic_upsample
(x, H, W)
models_search/Celeba256_gen.py:258
↓ 2 callersFunctioncalc_output_padding
(input_shape, output_shape)
torch_utils/ops/conv2d_gradfix.py:95
↓ 2 callersMethodclear
(self)
utils/utils.py:211
↓ 2 callersFunctionget_module_from_obj_name
Searches for the underlying module behind the name to some python object. Returns the module and the object name (original name with module part r
dnnlib/util.py:222
↓ 2 callersFunctionis_persistent
r"""Test whether the given object or class is persistent, i.e., whether it will save its source code when pickled.
torch_utils/persistence.py:134
↓ 2 callersMethodnames
r"""Returns the names of all statistics broadcasted so far that match the regular expression specified at construction time.
torch_utils/training_stats.py:141
↓ 2 callersFunctionnan_to_num
(input, nan=0.0, posinf=None, neginf=None, *, out=None)
torch_utils/misc.py:49
↓ 2 callersFunctionnorm_cdf
(x)
models_search/ViT_helper.py:60
↓ 2 callersFunctionnorm_ip
(img, low, high)
utils/utils.py:87
↓ 2 callersFunctionnorm_range
(t, value_range)
utils/utils.py:91
↓ 2 callersFunctionpixel_upsample
(x, H, W)
models_search/ViT_custom_scale2.py:242
↓ 2 callersFunctionpixel_upsample
(x, H, W)
models_search/Celeba256_dis.py:258
↓ 2 callersFunctionpixel_upsample
(x, H, W)
models_search/ViT_scale3_local_new_rp.py:242
↓ 2 callersFunctionpixel_upsample
(x, H, W)
models_search/ViT_custom_scale2_rp_noise.py:246
↓ 2 callersMethodset_arch
(self, x, cur_stage)
models_search/ViT_custom.py:269
↓ 2 callersFunctionset_log_dir
(root_dir, exp_name)
utils/utils.py:167
↓ 2 callersMethodstd
r"""Returns the standard deviation of the scalars that were accumulated for the given statistic between the last two calls to `update(
torch_utils/training_stats.py:198
↓ 2 callersFunctiontranslate2d_inv
(tx, ty, **kwargs)
models_search/ada.py:84
↓ 2 callersFunctionupdown
(x, H, W)
models_search/Celeba256_gen.py:269
↓ 2 callersFunctionvalidate
(args, fixed_z, fid_stat, epoch, gen_net: nn.Module, writer_dict, clean_dir=True)
functions.py:276
↓ 2 callersFunctionwindow_partition
Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
models_search/Celeba256_dis.py:185
↓ 2 callersFunctionwindow_partition
Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
models_search/Celeba256_gen.py:281
↓ 2 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
models_search/Celeba256_dis.py:199
↓ 2 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
models_search/Celeba256_gen.py:293
↓ 1 callersMethod__reduce__
(self)
torch_utils/persistence.py:118
↓ 1 callersMethod__setup_files
private helper for setting up the files_list :return: files => list of paths of files
celeba.py:21
↓ 1 callersMethod__setup_files
private helper for setting up the files_list :return: files => list of paths of files
celeba.py:84
↓ 1 callersFunction_bias_act_cuda
Fast CUDA implementation of `bias_act()` using custom ops.
torch_utils/ops/bias_act.py:129
↓ 1 callersFunction_bias_act_ref
Slow reference implementation of `bias_act()` using standard TensorFlow ops.
torch_utils/ops/bias_act.py:94
↓ 1 callersFunction_check_pickleable
r"""Check that the given object is pickleable, raising an exception if it is not. This function is expected to be considerably more efficient
torch_utils/persistence.py:231
↓ 1 callersFunction_find_compiler_bindir
()
torch_utils/custom_ops.py:28
↓ 1 callersFunction_get_no_grad_ctx_mgr
Returns a the `torch.no_grad` context manager for PyTorch version >= 0.4, or a no-op context manager otherwise.
utils/torch_fid_score.py:54
↓ 1 callersFunction_init
()
torch_utils/ops/bias_act.py:41
↓ 1 callersFunction_init
()
torch_utils/ops/upfirdn2d.py:26
↓ 1 callersFunction_init_inception
()
utils/inception_score.py:64
↓ 1 callersFunction_module_to_src
r"""Query the source code of a given Python module.
torch_utils/persistence.py:206
↓ 1 callersFunction_no_grad_trunc_normal_
(tensor, mean, std, a, b)
models_search/ViT_helper.py:57
↓ 1 callersFunction_should_use_custom_op
()
torch_utils/ops/grid_sample_gradfix.py:34
↓ 1 callersFunction_src_to_module
r"""Get or create a Python module for the given source code.
torch_utils/persistence.py:216
↓ 1 callersFunction_sync
r"""Synchronize the global cumulative counters across devices and processes. Called internally by `Collector.update()`.
torch_utils/training_stats.py:234
↓ 1 callersFunction_upfirdn2d_ref
Slow reference implementation of `upfirdn2d()` using standard PyTorch ops.
torch_utils/ops/upfirdn2d.py:169
↓ 1 callersFunctioncalculate_activation_statistics
Calculation of the statistics used by the FID. Params: -- images : Numpy array of dimension (n_images, hi, wi, 3). The values
utils/fid_score.py:173
next →1–100 of 560, ranked by callers