MCPcopy Create free account

hub / github.com/SageCao1125/SDM / functions

Functions158 in github.com/SageCao1125/SDM

↓ 17 callersFunctionextract
Extract some coefficients at specified timesteps, then reshape to [batch_size, 1, 1, 1, 1, ...] for broadcasting purposes.
SDM/diffusion.py:9
↓ 10 callersFunctionbuild_power_value
(B=2, additive=True)
ANN2SNN/quant_layer.py:15
↓ 10 callersFunctionextract
Extract some coefficients at specified timesteps, then reshape to [batch_size, 1, 1, 1, 1, ...] for broadcasting purposes.
ANN2SNN/diffusion.py:6
↓ 6 callersMethod__init__
(self, timestep, in_ch, out_ch, kernel_size=3, stride=1,padding=1)
SDM/TSM.py:18
↓ 6 callersMethod__init__
(self, in_features, out_features, bias=True)
ANN2SNN/quant_layer.py:365
↓ 5 callersFunctionapot_quantization
(tensor, alpha, proj_set, is_weight=True, grad_scale=None)
ANN2SNN/quant_layer.py:70
↓ 5 callersFunctionuniform_quantization
(tensor, alpha, bit, is_weight=True, grad_scale=None)
ANN2SNN/quant_layer.py:131
↓ 4 callersMethod__init__
(self, T, ch, ch_mult, num_res_blocks, dropout, bit=32, spike_time: int = 3)
ANN2SNN/spiking_model.py:133
↓ 4 callersMethod__init__
Build pretrained InceptionV3 Parameters ---------- output_blocks : list of int Indices of blocks to return featur
ANN2SNN/inception.py:32
↓ 4 callersMethod__init__
(self, T, ch, ch_mult, num_res_blocks, dropout,bit=32)
ANN2SNN/model.py:123
↓ 4 callersFunctionevaluate
(sampler, model)
ANN2SNN/ANN_train.py:80
↓ 4 callersFunctionextract2
Extract some coefficients at specified timesteps, then reshape to [batch_size, 1, 1, 1, 1, ...] for broadcasting purposes.
SDM/diffusion.py:17
↓ 4 callersFunctionget_inception_and_fid_score
when `images` is a python generator, `num_images` should be given
ANN2SNN/both.py:13
↓ 4 callersFunctionuq_with_calibrated_graditens
(grad_scale=None)
ANN2SNN/quant_layer.py:106
↓ 3 callersMethod__init__
(self)
ANN2SNN/spiking_layer.py:193
↓ 3 callersFunctionmodel_load_acta
(model, spiking = False)
ANN2SNN/SNN_ft.py:84
↓ 2 callersMethodbackward
(ctx, grad_output)
ANN2SNN/quant_layer.py:118
↓ 2 callersFunctionbypass_blocks
(model, para_ls)
ANN2SNN/SNN_ft.py:474
↓ 2 callersFunctioncalculate_frechet_distance
Numpy implementation of the Frechet Distance. The Frechet distance between two multivariate Gaussians X_1 ~ N(mu_1, C_1) and X_2 ~ N(mu_2, C_2
ANN2SNN/fid.py:66
↓ 2 callersFunctiongradient_scale
(x, scale)
ANN2SNN/quant_layer.py:63
↓ 2 callersMethodp_mean_variance
(self, x_t, t)
SDM/diffusion.py:152
↓ 2 callersFunctionpower_quant
(x, value_s)
ANN2SNN/quant_layer.py:71
↓ 2 callersMethodq_mean_variance
Compute the mean and variance of the diffusion posterior q(x_{t-1} | x_t, x_0)
SDM/diffusion.py:120
↓ 2 callersMethodq_mean_variance
Compute the mean and variance of the diffusion posterior q(x_{t-1} | x_t, x_0)
ANN2SNN/diffusion.py:88
↓ 2 callersFunctiontorch_cov
Estimate a covariance matrix given data. Covariance indicates the level to which two variables vary together. If we examine N-dimensional samp
ANN2SNN/fid.py:14
↓ 1 callersFunctionSNNsample
(seed)
SDM/sample.py:42
↓ 1 callersMethod__init__
(self, model, beta_1, beta_T, T)
SDM/diffusion.py:27
↓ 1 callersMethod__init__
(self, model, beta_1, beta_T, T)
ANN2SNN/diffusion.py:16
↓ 1 callersFunctionema
(source, target, decay)
ANN2SNN/ANN_train.py:61
↓ 1 callersFunctioneval
()
ANN2SNN/SNN_conversion.py:59
↓ 1 callersFunctioneval
()
ANN2SNN/ANN_train.py:226
↓ 1 callersFunctioneval
()
ANN2SNN/evaluate.py:68
↓ 1 callersFunctionevaluate
(sampler, model)
ANN2SNN/evaluate.py:115
↓ 1 callersFunctionfid_inception_v3
Build pretrained Inception model for FID computation The Inception model for FID computation uses a different set of weights and has a slight
ANN2SNN/inception.py:180
↓ 1 callersFunctionfine_tune
()
ANN2SNN/SNN_ft.py:177
↓ 1 callersFunctionget_statistics
when `images` is a python generator, `num_images` should be given
ANN2SNN/fid.py:145
↓ 1 callersFunctioninfiniteloop
(dataloader)
ANN2SNN/SNN_ft.py:145
↓ 1 callersFunctioninfiniteloop
(dataloader)
ANN2SNN/ANN_train.py:70
↓ 1 callersMethodinitialize
(self)
SDM/TSM.py:25
↓ 1 callersMethodinitialize
(self)
SDM/TSM.py:67
↓ 1 callersMethodinitialize
(self)
SDM/TSM.py:171
↓ 1 callersMethodinitialize
(self)
ANN2SNN/spiking_model.py:30
↓ 1 callersMethodinitialize
(self)
ANN2SNN/spiking_model.py:48
↓ 1 callersMethodinitialize
(self)
ANN2SNN/spiking_model.py:71
↓ 1 callersMethodinitialize
(self)
ANN2SNN/spiking_model.py:113
↓ 1 callersMethodinitialize
(self)
ANN2SNN/spiking_model.py:183
↓ 1 callersMethodinitialize
(self)
ANN2SNN/model.py:30
↓ 1 callersMethodinitialize
(self)
ANN2SNN/model.py:48
↓ 1 callersMethodinitialize
(self)
ANN2SNN/model.py:66
↓ 1 callersMethodinitialize
(self)
ANN2SNN/model.py:104
↓ 1 callersMethodinitialize
(self)
ANN2SNN/model.py:173
↓ 1 callersFunctionmain
()
SDM/sample.py:82
↓ 1 callersFunctionmodel_update
(agent, target)
ANN2SNN/SNN_ft.py:67
↓ 1 callersMethodp_mean_variance
(self, x_t, t)
ANN2SNN/diffusion.py:119
↓ 1 callersFunctionpara_ls_make
()
ANN2SNN/SNN_ft.py:107
↓ 1 callersMethodpredict_xstart_from_eps
(self, x_t, t, eps)
SDM/diffusion.py:134
↓ 1 callersMethodpredict_xstart_from_eps
(self, x_t, t, eps)
ANN2SNN/diffusion.py:102
↓ 1 callersMethodpredict_xstart_from_xprev
(self, x_t, t, xprev)
SDM/diffusion.py:142
↓ 1 callersMethodpredict_xstart_from_xprev
(self, x_t, t, xprev)
ANN2SNN/diffusion.py:109
↓ 1 callersFunctionseed_everything
(seed_value)
SDM/sample.py:30
↓ 1 callersFunctionsqrt_newton_schulz
(A, numIters, dtype=None)
ANN2SNN/fid.py:46
↓ 1 callersFunctionswitch_on
(model, para_ls)
ANN2SNN/SNN_ft.py:452
↓ 1 callersFunctiontrain
()
ANN2SNN/ANN_train.py:98
Method__init__
(self, model, beta_1, beta_T, T, img_size=32, mean_type='xstart', var_type='fixedlarge',sampl
SDM/diffusion.py:60
Method__init__
(self, T, d_model, dim)
SDM/TSM.py:47
Method__init__
(self, in_ch, timestep=4, global_thres=1.0, use_cupy=False)
SDM/TSM.py:79
Method__init__
(self, in_ch, timestep=4, global_thres=1.0, use_cupy=False)
SDM/TSM.py:106
Method__init__
(self, in_ch, out_ch, tdim, dropout, attn=False, timestep=4, global_thres=1.0, use_cupy=False)
SDM/TSM.py:135
Method__init__
(self, timestep=4)
SDM/TSM.py:207
Method__init__
(self, T, ch, ch_mult, attn, num_res_blocks, dropout, timestep, img_ch=3, global_thres=1.0, use_cupy=False)
SDM/TSM.py:223
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bia
ANN2SNN/quant_layer.py:167
Method__init__
(self, inplace: bool = False, bit=5, power=False, additive=True, grad_scale=None)
ANN2SNN/quant_layer.py:211
Method__init__
(self, inplace: bool = False, bit=5, power=False, additive=True, grad_scale=None)
ANN2SNN/quant_layer.py:246
Method__init__
(self, kernel_size, stride, bit=5, power=False, additive=True, grad_scale=None)
ANN2SNN/quant_layer.py:278
Method__init__
(self, in_features, out_features, bias=True, bit=5, power=True, additive=True, grad_scale=None)
ANN2SNN/quant_layer.py:310
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=False)
ANN2SNN/quant_layer.py:351
Method__init__
(self, block, T, alpha_loc=2)
ANN2SNN/spiking_layer.py:14
Method__init__
(self, block, T, alpha_loc)
ANN2SNN/spiking_layer.py:72
Method__init__
(self, block, T, alpha_loc)
ANN2SNN/spiking_layer.py:135
Method__init__
(self, T, d_model, dim, bit, spike_time: int = 3)
ANN2SNN/spiking_model.py:10
Method__init__
(self, in_ch)
ANN2SNN/spiking_model.py:42
Method__init__
(self, in_ch)
ANN2SNN/spiking_model.py:65
Method__init__
(self, in_ch, out_ch, tdim, dropout, bit, spike_time: int = 3)
ANN2SNN/spiking_model.py:89
Method__init__
(self, model, beta_1, beta_T, T, img_size=32, mean_type='eps', var_type='fixedlarge')
ANN2SNN/diffusion.py:47
Method__init__
(self, in_channels, pool_features)
ANN2SNN/inception.py:209
Method__init__
(self, in_channels, channels_7x7)
ANN2SNN/inception.py:234
Method__init__
(self, in_channels)
ANN2SNN/inception.py:262
Method__init__
(self, in_channels)
ANN2SNN/inception.py:295
Method__init__
(self, T, d_model, dim, bit)
ANN2SNN/model.py:10
Method__init__
(self, in_ch)
ANN2SNN/model.py:42
Method__init__
(self, in_ch)
ANN2SNN/model.py:60
Method__init__
(self, in_ch, out_ch, tdim, dropout,bit)
ANN2SNN/model.py:80
Functionadjust_learning_rate
For resnet, the lr starts from 0.1, and is divided by 10 at 80 and 120 epochs
ANN2SNN/SNN_ft.py:60
Functionema
(source, target, decay)
ANN2SNN/SNN_ft.py:136
Functionevaluate
(sampler, model)
ANN2SNN/SNN_conversion.py:42
Functionevaluate
(sampler, model)
ANN2SNN/SNN_ft.py:155
Methodextra_repr
(self)
ANN2SNN/spiking_layer.py:206
Methodforward
Algorithm 1.
SDM/diffusion.py:44
Methodforward
(self, x_T)
SDM/diffusion.py:231
Methodforward
(self, input)
SDM/TSM.py:29
next →1–100 of 158, ranked by callers