MCPcopy Create free account

hub / github.com/ChangyuanWang17/APQ-DM / functions

Functions210 in github.com/ChangyuanWang17/APQ-DM

Method__init__
(self, window_size=20, fmt=None)
util.py:23
Method__init__
(self, delimiter="\t")
util.py:81
Method__init__
(self)
utils/compute.py:73
Method__init__
(self, in_channels=128, out_channels=128, w_bit=8, a_bit=8, half_wave=False, sequence=None, args=None)
utils/quant_util.py:71
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bia
utils/quant_util.py:348
Method__init__
(self, length, size, transform)
utils/data_utils.py:29
Method__init__
(self, args, config, device=None)
runners/diffusion.py:67
Method__init__
(self, files, transforms=None)
pytorch-fid-master/src/pytorch_fid/fid_score.py:78
Method__init__
(self, in_channels, pool_features)
pytorch-fid-master/src/pytorch_fid/inception.py:226
Method__init__
(self, in_channels, channels_7x7)
pytorch-fid-master/src/pytorch_fid/inception.py:251
Method__init__
(self, in_channels)
pytorch-fid-master/src/pytorch_fid/inception.py:279
Method__init__
(self, in_channels)
pytorch-fid-master/src/pytorch_fid/inception.py:312
Method__init__
(self, path, transform, resolution=8)
datasets/ffhq.py:9
Method__init__
(self, root, transforms=None, transform=None, target_transform=None)
datasets/vision.py:9
Method__init__
(self, transform=None, target_transform=None)
datasets/vision.py:59
Method__init__
(self, x1, x2, y1, y2)
datasets/__init__.py:15
Method__init__
(self, root, transform=None, target_transform=None)
datasets/lsun.py:12
Method__init__
(self, root, split="train", target_type="attr", transform=N
datasets/celeba.py:50
Method__init__
(self, mu=0.999)
models/ema.py:5
Method__init__
(self, in_channels, with_conv)
models/diffusion.py:39
Method__init__
(self, in_channels, with_conv)
models/diffusion.py:58
Method__init__
(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout, temb_channels=512, qu
models/diffusion.py:80
Method__init__
(self, in_channels)
models/diffusion.py:169
Method__len__
(self)
utils/data_utils.py:34
Method__len__
(self)
pytorch-fid-master/src/pytorch_fid/fid_score.py:82
Method__len__
(self)
datasets/ffhq.py:28
Method__len__
(self)
datasets/lsun.py:57
Method__len__
(self)
datasets/lsun.py:171
Method__len__
(self)
datasets/celeba.py:158
Method__repr__
(self)
datasets/vision.py:75
Method__repr__
(self)
datasets/__init__.py:24
Method__str__
(self)
util.py:71
Method__str__
(self)
util.py:100
Function_load_checkpoint_for_ema
Workaround for ModelEma._load_checkpoint to accept an already-loaded object
util.py:162
Method_quantize_bias
(self, bias)
utils/quant_util.py:301
Methoda_bit
(self)
utils/quant_util.py:133
Methodadd_meter
(self, name, meter)
util.py:112
Methodavg
(self)
util.py:55
Methodb_bit
(self)
utils/quant_util.py:141
Methodbackward
(ctx, grad_output)
utils/quantization_utils/quant_utils.py:165
Functionbar_update
(count, block_size, total_size)
datasets/utils.py:11
Methodcalibrate
(self, model, image, device)
runners/diffusion.py:584
Functionclamp
Clamp tensor input to (min, max). input: input tensor to be clamped
utils/quantization_utils/quant_utils.py:50
Functioncompute_singlecrop
(outputs, labels, loss, top5_flag=False, mean_flag=False)
utils/compute.py:31
Functioncompute_tencrop
(outputs, labels)
utils/compute.py:9
Functiondevice
()
pytorch-fid-master/tests/test_fid_score.py:10
Functiondownload_url
Download a file from a url and place it in root. Args: url (str): URL to download file from root (str): Directory to place downlo
datasets/utils.py:49
Functiondummy_statistics
(path, model, batch_size, dims, device, num_workers)
pytorch-fid-master/tests/test_fid_score.py:19
Methodema_copy
(self, module)
models/ema.py:31
Methodextra_repr
(self)
utils/quant_util.py:329
Methodextra_repr
(self)
utils/quant_util.py:383
Methodextra_repr
(self)
datasets/lsun.py:174
Methodextra_repr
(self)
datasets/celeba.py:161
Functionfind_MSESmallest
(x, k, x_min=None, x_max=None)
utils/quantization_utils/quant_utils.py:36
Functionfind_scale_by_percentile_max
(x, percentile=0.9999)
utils/quant_util.py:442
Functionfind_scale_by_percentile_min
(x, percentile=0.9999)
utils/quant_util.py:436
Methodforward
(self, inputs, a_bit)
utils/quant_util.py:54
Methodforward
(self, *inputs)
utils/quant_util.py:326
Methodforward
(ctx, origin_inputs, wanted_inputs)
utils/quant_util.py:339
Methodforward
(self, inputs)
utils/quant_util.py:379
Methodforward
x: single-precision value to be quantized k: bit-setting for x x_min: lower bound for quantization range x_max=None
utils/quantization_utils/quant_utils.py:142
Methodforward
Get Inception feature maps Parameters ---------- inp : torch.autograd.Variable Input tensor of shape Bx3xHxW. Val
pytorch-fid-master/src/pytorch_fid/inception.py:129
Methodforward
(self, x)
pytorch-fid-master/src/pytorch_fid/inception.py:229
Methodforward
(self, x)
pytorch-fid-master/src/pytorch_fid/inception.py:254
Methodforward
(self, x)
pytorch-fid-master/src/pytorch_fid/inception.py:282
Methodforward
(self, x)
pytorch-fid-master/src/pytorch_fid/inception.py:315
Methodforward
(self, x)
models/diffusion.py:49
Methodforward
(self, x)
models/diffusion.py:69
Methodforward
(self, x, temb)
models/diffusion.py:146
Methodforward
(self, x)
models/diffusion.py:195
Methodforward
(self, x, t)
models/diffusion.py:344
FunctiongetRandomData
get random sample dataloader dataset: name of the dataset batch_size: the batch size of random data for_inception: whether the data
utils/data_utils.py:44
FunctiongetTestData
Get dataloader of testset dataset: name of the dataset batch_size: the batch size of random data path: the path to the data for
utils/data_utils.py:76
Functionget_world_size
()
util.py:195
Methodglobal_avg
(self)
util.py:60
Methodhalf_wave
(self)
utils/quant_util.py:145
Functioninit_distributed_mode
(args)
util.py:216
Functionkmeans_update_model
(model, quantizable_idx, centroid_label_dict, free_high_bit=False)
utils/quant_util.py:20
Functionlint
(session)
pytorch-fid-master/noxfile.py:7
Functionlist_dir
List all directories at a given root Args: root (str): Path to directory whose folders need to be listed prefix (bool, optional):
datasets/utils.py:88
Functionlist_files
List all files ending with a suffix at a given root Args: root (str): Path to directory whose folders need to be listed suffix (s
datasets/utils.py:110
Methodlog_every
(self, iterable, print_freq, header=None)
util.py:115
Functionlp_loss
loss function measured in L_p Norm
utils/quantization_utils/quant_utils.py:26
Methodmedian
(self)
util.py:50
Functionnoise_estimation_loss
(model, x0: torch.Tensor, t: torch.LongTensor,
functions/losses.py:4
Methodquantized
(self)
utils/quant_util.py:149
Functionreconstruct_weight_from_k_means_result
(centroids, labels)
utils/quant_util.py:14
Functionsave_on_master
(*args, **kwargs)
util.py:211
Methodset_activation_range
(self, activation_range)
utils/quant_util.py:167
Methodset_fix_weight
(self, fix_weight)
utils/quant_util.py:164
Methodset_quantize
(self, quantized)
utils/quant_util.py:156
Methodset_tanh
(self, tanh=True)
utils/quant_util.py:183
Methodset_tanh_weight
(self, tanh_weight)
utils/quant_util.py:159
Methodset_trainable_activation_range
(self, trainable_activation_range=True)
utils/quant_util.py:173
Methodset_weight_range
(self, weight_range)
utils/quant_util.py:170
Methodslerp
(z1, z2, alpha)
runners/diffusion.py:497
Methodsynchronize_between_processes
Warning: does not synchronize the deque!
util.py:36
Methodsynchronize_between_processes
(self)
util.py:108
Methodtanh_weight
(self)
utils/quant_util.py:153
Functiontest
test a model on a given dataset
utils/train_utils.py:27
← previousnext →101–200 of 210, ranked by callers