MCPcopy Create free account

hub / github.com/NVlabs/stylegan2-ada / functions

Functions336 in github.com/NVlabs/stylegan2-ada

↓ 2 callersFunctiontranslate_2d
(tx, ty)
training/augment.py:186
↓ 2 callersFunctionupsample_2d
r"""Upsample a batch of 2D images with the given filter. Accepts a batch of 2D images of the shape `[N, C, H, W]` or `[N, H, W, C]` and upsam
dnnlib/tflib/ops/upfirdn_2d.py:176
↓ 1 callersMethod__enter__
(self)
dnnlib/util.py:72
↓ 1 callersMethod__init__
(self, num_threads)
dataset_tool.py:237
↓ 1 callersMethod_broadcast_nccl
Sum gradients across devices using NCCL ops (fast path).
dnnlib/tflib/optimizer.py:289
↓ 1 callersMethod_evaluate
(self, **_kwargs)
metrics/metric_base.py:90
↓ 1 callersFunction_find_compiler_bindir
()
dnnlib/tflib/custom_ops.py:35
↓ 1 callersFunction_fused_bias_act_ref
Slow reference implementation of `fused_bias_act()` using standard TensorFlow ops.
dnnlib/tflib/ops/fused_bias_act.py:76
↓ 1 callersFunction_get_compute_cap
(device)
dnnlib/tflib/custom_ops.py:50
↓ 1 callersFunction_get_cuda_gpu_arch_string
()
dnnlib/tflib/custom_ops.py:57
↓ 1 callersFunction_get_plugin
()
dnnlib/tflib/ops/upfirdn_2d.py:16
↓ 1 callersFunction_get_plugin
()
dnnlib/tflib/ops/fused_bias_act.py:17
↓ 1 callersFunction_handle_legacy_output_transforms
(output_transform, dynamic_kwargs)
dnnlib/tflib/network.py:746
↓ 1 callersMethod_increment_acc
(self, name, expr)
training/augment.py:138
↓ 1 callersMethod_read_and_decay_acc
(self, name, nimg_delta)
training/augment.py:151
↓ 1 callersMethod_set_strength
(self, strength)
training/augment.py:132
↓ 1 callersMethodadd_task
(self, func, args=())
dataset_tool.py:247
↓ 1 callersMethodapply
(self, images, labels, enable=True)
training/augment.py:90
↓ 1 callersMethodapply_gradients
(self, grads_and_vars)
dnnlib/tflib/optimizer.py:344
↓ 1 callersMethodapply_loss_scaling
Apply dynamic loss scaling for the given expression.
dnnlib/tflib/optimizer.py:275
↓ 1 callersFunctionapply_spectral_norm
(w, state_var='sn', iterations=1, eps=1e-8)
training/networks.py:168
↓ 1 callersFunctionbatch_pairwise_distances
Compute pairwise distances between two batches of feature vectors.
metrics/precision_recall.py:23
↓ 1 callersFunctioncalc_metrics
(network_pkl, metric_names, metricdata, mirror, gpus)
calc_metrics.py:27
↓ 1 callersFunctioncall_func_by_name
Finds the python object with the given name and calls it as a function.
dnnlib/util.py:276
↓ 1 callersMethodclose
(self)
dataset_tool.py:60
↓ 1 callersMethodcompute_gradients
(self, loss, var_list, gate_gradients=tf.train.Optimizer.GATE_NONE)
dnnlib/tflib/optimizer.py:340
↓ 1 callersFunctioncompute_kid
(feat_real, feat_fake, num_subsets=100, max_subset_size=1000)
metrics/kernel_inception_distance.py:23
↓ 1 callersFunctionconditional_entropy
(p)
metrics/linear_separability.py:98
↓ 1 callersMethodconfigure
(self, dataset_args={}, run_dir=None, progress_fn=None)
metrics/metric_base.py:43
↓ 1 callersFunctionconv_downsample_2d
r"""Fused `tf.nn.conv2d()` followed by `downsample_2d()`. Padding is performed only once at the beginning, not between the operations. The fu
dnnlib/tflib/ops/upfirdn_2d.py:313
↓ 1 callersMethodconvert
Create new network with the given parameters, and copy all variables from this network.
dnnlib/tflib/network.py:518
↓ 1 callersMethodcopy_own_vars_from
Copy the values of all variables from the given network, excluding sub-networks.
dnnlib/tflib/network.py:460
↓ 1 callersFunctioncreate_session
Create tf.Session based on config dict.
dnnlib/tflib/tfutil.py:151
↓ 1 callersFunctiondownsample
(y)
training/networks.py:584
↓ 1 callersFunctionentropy
(p)
metrics/linear_separability.py:88
↓ 1 callersFunctionexecute_cmdline
(argv)
dataset_tool.py:861
↓ 1 callersFunctionfinalize_autosummaries
Create the necessary ops to include autosummaries in TensorBoard report. Note: This should be done only once per graph.
dnnlib/tflib/autosummary.py:120
↓ 1 callersMethodfinish
(self)
dataset_tool.py:260
↓ 1 callersFunctionfunc_nonzero_2nd_grad
(x, b)
dnnlib/tflib/ops/fused_bias_act.py:191
↓ 1 callersFunctionfunc_zero_2nd_grad
(x, b)
dnnlib/tflib/ops/fused_bias_act.py:177
↓ 1 callersFunctionfused_bias_act
r"""Fused bias and activation function. Adds bias `b` to activation tensor `x`, evaluates activation function `act`, and scales the result by
dnnlib/tflib/ops/fused_bias_act.py:36
↓ 1 callersFunctiongenerate_images
(network_pkl, seeds, truncation_psi, outdir, class_idx, dlatents_npz)
generate.py:24
↓ 1 callersFunctionget_obj_by_name
Finds the python object with the given name.
dnnlib/util.py:270
↓ 1 callersMethodget_result
(self, func)
dataset_tool.py:253
↓ 1 callersMethodget_result_str
(self)
metrics/metric_base.py:77
↓ 1 callersMethodget_var
Get the value of a given variable as NumPy array. Note: This method is very inefficient -- prefer to use tflib.run(list_of_vars) whenever poss
dnnlib/tflib/network.py:393
↓ 1 callersFunctiongrad2_d_x
(d_dx, d_db, x, y)
dnnlib/tflib/ops/fused_bias_act.py:170
↓ 1 callersFunctiongrad_impl
(dy, x)
dnnlib/tflib/ops/fused_bias_act.py:195
↓ 1 callersMethodinit_validation_set
(self, D_gpus, training_set)
training/augment.py:67
↓ 1 callersFunctionis_top_level_function
Determine whether the given object is a top-level function, i.e., defined at module scope using 'def'.
dnnlib/util.py:295
↓ 1 callersFunctionis_url
Determine whether the given object is a valid URL string.
dnnlib/util.py:361
↓ 1 callersFunctionknn_precision_recall_features
Calculates k-NN precision and recall for two sets of feature vectors.
metrics/precision_recall.py:141
↓ 1 callersMethodlist_layers
Returns a list of (layer_name, output_expr, trainable_vars) tuples corresponding to individual layers of the network. Mainly intended to be us
dnnlib/tflib/network.py:651
↓ 1 callersMethodlist_ops
(self)
dnnlib/tflib/network.py:642
↓ 1 callersMethodload_validation_set_np
(self)
training/dataset.py:183
↓ 1 callersFunctionmain
()
train.py:509
↓ 1 callersFunctionmain
()
calc_metrics.py:138
↓ 1 callersFunctionmain
()
style_mixing.py:98
↓ 1 callersFunctionmain
()
generate.py:100
↓ 1 callersFunctionmain
()
projector.py:270
↓ 1 callersFunctionmake_cache_dir_path
(*paths: str)
dnnlib/util.py:121
↓ 1 callersFunctionminibatch_stddev_layer
(x, group_size=None, num_new_features=1)
training/networks.py:142
↓ 1 callersFunctionmutual_information
(p)
metrics/linear_separability.py:74
↓ 1 callersMethodparse_tfrecord_np
(record)
training/dataset.py:215
↓ 1 callersFunctionproject
(network_pkl: str, target_fname: str, outdir: str, save_video: bool, seed: int)
projector.py:205
↓ 1 callersMethodreport_stat
(self, name, expr)
training/augment.py:103
↓ 1 callersFunctionrotate_2d
(theta)
training/augment.py:212
↓ 1 callersFunctionrotate_3d
(v, theta)
training/augment.py:218
↓ 1 callersFunctionrun_training
(outdir, seed, dry_run, **hyperparam_options)
train.py:413
↓ 1 callersMethodrun_validation
(self, minibatch_size)
training/augment.py:126
↓ 1 callersFunctionscale_3d
(sx, sy, sz)
training/augment.py:205
↓ 1 callersMethodset_network
(self, Gs, dtype='float16')
projector.py:62
↓ 1 callersFunctionset_vars
Set the values of given tf.Variables. Equivalent to the following, but more efficient and does not bloat the tf graph: tflib.run([tf.assign(v
dnnlib/tflib/tfutil.py:204
↓ 1 callersMethodsetup_as_moving_average_of
Construct a TensorFlow op that updates the variables of this network to be slightly closer to those of the given network.
dnnlib/tflib/network.py:528
↓ 1 callersFunctionsetup_snapshot_image_grid
(training_set)
training/training_loop.py:27
↓ 1 callersFunctionsetup_training_options
( # General options (not included in desc). gpus = None, # Number of GPUs: <int>, default = 1 gp
train.py:31
↓ 1 callersMethodstep
(self)
projector.py:163
↓ 1 callersFunctionstyle_mixing_example
(network_pkl, row_seeds, col_seeds, truncation_psi, col_styles, outdir, minibatch_size=4)
style_mixing.py:24
↓ 1 callersFunctiontranslate_3d
(tx, ty, tz)
training/augment.py:192
↓ 1 callersMethodtune
(self, nimg_delta)
training/augment.py:108
↓ 1 callersMethodundo_loss_scaling
Undo the effect of dynamic loss scaling for the given expression.
dnnlib/tflib/optimizer.py:282
↓ 1 callersMethodupdate_autosummaries
(self)
metrics/metric_base.py:86
↓ 1 callersFunctionupsample
(y)
training/networks.py:417
↓ 1 callersFunctionupsample_conv_2d
r"""Fused `upsample_2d()` followed by `tf.nn.conv2d()`. Padding is performed only once at the beginning, not between the operations. The fuse
dnnlib/tflib/ops/upfirdn_2d.py:245
↓ 1 callersMethodvariables
(self)
dnnlib/tflib/optimizer.py:337
FunctionD_main
( images_in, # First input: Images [minibatch, channel, height, width]. label
training/networks.py:459
FunctionG_main
( latents_in, # First input: Latent vectors (Z) [minibatch, latent_siz
training/networks.py:188
FunctionG_mapping
( latents_in, # First input: Latent vectors (Z) [minibatch, latent_size].
training/networks.py:283
FunctionG_synthesis
( dlatents_in, # Input: Disentangled latents (W) [minibatch, num_layers, dlatent_si
training/networks.py:343
Method__delattr__
(self, name: str)
dnnlib/util.py:52
Method__enter__
(self)
dataset_tool.py:107
Method__enter__
(self)
dataset_tool.py:203
Method__enter__
(self)
dataset_tool.py:264
Method__exit__
(self, *args)
dataset_tool.py:110
Method__exit__
(self, *args)
dataset_tool.py:206
Method__exit__
(self, *excinfo)
dataset_tool.py:267
Method__exit__
(self, exc_type: Any, exc_value: Any, traceback: Any)
dnnlib/util.py:75
Method__getattr__
(self, name: str)
dnnlib/util.py:43
Method__getstate__
Pickle export.
dnnlib/tflib/network.py:403
Method__init__
(self, tfrecord_dir, expected_images, print_progress=True, progress_interval=10, tfr_prefix=None)
dataset_tool.py:39
← previousnext →101–200 of 336, ranked by callers