Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/CVL-UESTC/MVAR
/ functions
Functions
241 in github.com/CVL-UESTC/MVAR
⨍
Functions
241
◇
Types & classes
45
↓ 1 callers
Method
remaining
(self)
utils/evaluations/c2i/evaluator.py:445
↓ 1 callers
Function
sample_with_top_k_top_p_
(logits_BlV: torch.Tensor, top_k: int = 0, top_p: float = 0.0, rng=None, num_samples=1)
models/helpers.py:6
↓ 1 callers
Method
seed_everything
(self, benchmark: bool)
utils/arg_util.py:142
↓ 1 callers
Method
set_tf32
(tf32: bool)
utils/arg_util.py:215
↓ 1 callers
Function
slow_attn
(query, key, value, scale: float, attn_mask=None, dropout_p=0.0)
models/basic_mvar.py:40
↓ 1 callers
Method
synchronize_between_processes
(self)
utils/misc.py:347
↓ 1 callers
Method
time_preds
(self, counts)
utils/misc.py:296
↓ 1 callers
Method
train_one_ep_ratio_k
( self, *, ep: int, is_first_ep: bool, start_it: int, args: ar
trainer.py:286
↓ 1 callers
Method
train_step
( self, *, it: int, g_it: int, stepping: bool, metric_lg: Metr
trainer.py:203
↓ 1 callers
Method
update_patch_related
(self, x_BlC, attn_bias)
models/mvar.py:316
↓ 1 callers
Method
warmup
(self)
utils/evaluations/c2i/evaluator.py:141
Method
__call__
(self, pil_image)
utils/data.py:77
Method
__del__
(self)
utils/misc.py:170
Method
__enter__
(self)
utils/amp_sc.py:8
Method
__exit__
(self, exc_type, exc_val, exc_tb)
utils/amp_sc.py:11
Method
__getattr__
(self, attr: str)
utils/misc.py:182
Method
__getattr__
(self, attr)
utils/misc.py:331
Method
__getitem__
Args: index (int): Index Returns: tuple: (sample, target, filename).
utils/data.py:15
Method
__getitem__
Args: index (int): Index Returns: tuple: (moments, target).
utils/data.py:46
Method
__getitem__
(self, idx)
utils/data.py:163
Method
__getitem__
(self, at_from_0_to_1: float)
models/quant.py:369
Method
__getitem__
(self, at_from_0_to_1: float)
models/quant.py:387
Method
__init__
(self, module, *args, **kwargs)
train.py:451
Method
__init__
( self, *, device, patch_nums: Tuple[int, ...], resos: Tuple[int, ...]
trainer.py:32
Method
__init__
(self, dataset, num_replicas, rank)
utils/data_sampler.py:7
Method
__init__
( self, dataset_len, batch_size, seed_for_all_rank=0,
utils/data_sampler.py:21
Method
__init__
( self, world_size, rank, dataset_len, glb_bat
utils/data_sampler.py:77
Method
__init__
(self, local_output_dir, sync_stdout=True)
utils/misc.py:134
Method
__init__
(self, lg, verbose)
utils/misc.py:175
Method
__init__
(self, log_dir, filename_suffix)
utils/misc.py:187
Method
__init__
(self, window_size=30, fmt=None)
utils/misc.py:252
Method
__init__
(self, delimiter=" ")
utils/misc.py:315
Method
__init__
( self, root: str, )
utils/data.py:36
Method
__init__
(self, img_size)
utils/data.py:74
Method
__init__
(self, label_list)
utils/data.py:157
Method
__init__
( self, mixed_precision: int, optimizer: torch.optim.Optimizer, names: List[s
utils/amp_sc.py:16
Method
__init__
(self, mu: np.ndarray, sigma: np.ndarray)
utils/evaluations/c2i/evaluator.py:74
Method
__init__
( self, session, batch_size=64, softmax_batch_size=512, )
utils/evaluations/c2i/evaluator.py:125
Method
__init__
Estimate the manifold of given feature vectors. :param session: the TensorFlow session. :param row_batch_size: row batch siz
utils/evaluations/c2i/evaluator.py:218
Method
__init__
(self, session)
utils/evaluations/c2i/evaluator.py:375
Method
__init__
(self, gen_fn, length)
utils/evaluations/c2i/evaluator.py:462
Method
__init__
(self, arr_f, shape, dtype)
utils/evaluations/c2i/evaluator.py:474
Method
__init__
(self, arr)
utils/evaluations/c2i/evaluator.py:500
Method
__init__
( self, vocab_size, Cvae, using_znorm, beta: float = 0.25, def
models/quant.py:19
Method
__init__
(self, qresi: Phi)
models/quant.py:350
Method
__init__
(self, qresi_ls: nn.ModuleList)
models/quant.py:359
Method
__init__
(self, qresi: List)
models/quant.py:377
Method
__init__
( self, # VQVAE args vae_local: VQVAE, # MVAR args num_classes=1000,
models/mvar.py:39
Method
__init__
(self, in_channels)
models/basic_vae.py:30
Method
__init__
(self, in_channels)
models/basic_vae.py:41
Method
__init__
( self, *, in_channels, out_channels=None, dropout )
models/basic_vae.py:52
Method
__init__
(self, in_channels)
models/basic_vae.py:83
Method
__init__
( self, *, ch=128, ch_mult=(1, 2, 4, 8), num_res_blocks=2, dro
models/basic_vae.py:213
Method
__init__
(self, drop_prob: float = 0., scale_by_keep: bool = True)
models/helpers.py:50
Method
__init__
( self, block_idx, embed_dim=768, num_heads=12, attn_drop=0.0,
models/basic_mvar.py:129
Method
__init__
( self, block_idx, last_drop_p, embed_dim, cond_dim, shared_al
models/basic_mvar.py:264
Method
__init__
(self, C, D, norm_layer)
models/basic_mvar.py:352
Method
__init__
( self, vocab_size=4096, z_channels=32, ch=128, dropout=0.0, b
models/vqvae.py:18
Method
__iter__
(self)
utils/data_sampler.py:13
Method
__iter__
(self)
utils/data_sampler.py:60
Method
__iter__
(self)
utils/evaluations/c2i/evaluator.py:469
Method
__len__
(self)
utils/data_sampler.py:16
Method
__len__
(self)
utils/data_sampler.py:72
Method
__len__
(self)
utils/evaluations/c2i/evaluator.py:466
Method
__str__
(self)
utils/misc.py:304
Method
__str__
(self)
utils/misc.py:340
Method
__str__
(self)
utils/arg_util.py:271
Function
allgather
(t: torch.Tensor, cat=True)
utils/dist.py:108
Function
allgather_diff_shape
(t: torch.Tensor, cat=True)
utils/dist.py:121
Method
avg
(self)
utils/misc.py:281
Function
broadcast
(t: torch.Tensor, src_rank)
utils/dist.py:148
Method
close
(self)
utils/misc.py:243
Function
compile_model
(m, fast)
main_cache.py:58
Function
dist_fmt_vals
(val: float, fmt: Union[str, None] = '%.2f')
utils/dist.py:158
Method
do_nothing
(*args, **kwargs)
utils/misc.py:179
Function
echo
(info)
utils/misc.py:50
Method
eini
(self, eini)
models/quant.py:72
Method
evaluate
Evaluate if new feature vectors are at the manifold.
utils/evaluations/c2i/evaluator.py:289
Method
extra_repr
(self)
models/quant.py:80
Method
extra_repr
(self)
models/quant.py:372
Method
extra_repr
(self)
models/quant.py:392
Method
extra_repr
(self)
models/mvar.py:518
Method
extra_repr
(self)
models/helpers.py:58
Method
extra_repr
(self)
models/basic_mvar.py:124
Method
extra_repr
(self)
models/basic_mvar.py:258
Method
extra_repr
(self)
models/basic_mvar.py:347
Function
finalize
()
utils/dist.py:208
Method
flush
(self)
utils/misc.py:240
Function
for_visualize
(func)
utils/dist.py:196
Method
forward
(self, *args, **kwargs)
train.py:456
Method
forward
H, W is the height and width after encoding
models/quant.py:84
Method
forward
(self, h_BChw)
models/quant.py:343
Method
forward
(self, cond_BD)
models/mvar.py:32
Method
forward
(self, x)
models/basic_vae.py:36
Method
forward
(self, x)
models/basic_vae.py:47
Method
forward
(self, x)
models/basic_vae.py:76
Method
forward
(self, x)
models/basic_vae.py:96
Method
forward
(self, x)
models/basic_vae.py:193
Method
forward
(self, z)
models/basic_vae.py:278
Method
forward
(self, x)
models/helpers.py:55
← previous
next →
101–200 of 241, ranked by callers