MCPcopy Create free account

hub / github.com/Vermeille/Torchelie / functions

Functions1,230 in github.com/Vermeille/Torchelie

Method__getitem__
(self, i: int)
torchelie/datasets/concat.py:80
Method__getitem__
(self, i: int)
torchelie/datasets/concat.py:103
Method__getitem__
(self, i)
torchelie/datasets/concat.py:126
Method__getitem__
(self, i)
tests/test_datasets.py:13
Method__getitem__
(self, i)
tests/test_recipes.py:20
Method__getitem__
(self, i)
tests/test_recipes.py:30
Method__getitem__
(self, item)
tests/test_tensorboard_callback.py:28
Method__getitem__
(self, i)
examples/conditional.py:41
Function__getstate__
(self)
torchelie/utils.py:728
Method__init__
(self, in_channels: int, hidden_channels: int)
wip/singan.py:10
Method__init__
(self, in_channels: int, hidden_channels: int)
wip/singan.py:38
Method__init__
(self, name, fmt=':f')
wip/imagenet.py:433
Method__init__
(self, num_batches, meters, prefix="")
wip/imagenet.py:460
Method__init__
(self, path, l=2, ctfm=None, stfm=None)
wip/adain.py:96
Method__init__
(self)
wip/adain.py:134
Method__init__
(self, root: str, classes: List[str], capacity: int = 200)
torchelie/serving_utils.py:16
Method__init__
(self, params, lr=1e-3, eps=1e-8, weight_decay=0)
torchelie/optim.py:19
Method__init__
(self, params, lr=1e-3, beta=0.9, weight_decay=0)
torchelie/optim.py:66
Method__init__
(self, params, lr=1e-3, betas=(0.9, 0.999),
torchelie/optim.py:126
Method__init__
(self, params, lr=1e-3, betas=(0.9, 0.999),
torchelie/optim.py:230
Method__init__
(self, base_optimizer, alpha=0.5, k=5)
torchelie/optim.py:327
Method__init__
(self, m: nn.Module)
torchelie/utils.py:431
Method__init__
(self, m)
torchelie/utils.py:452
Method__init__
(self, except_names: List[str] = [])
torchelie/utils.py:586
Method__init__
(self, shape: Tuple[int, ...], sd: float = 0.01, decay_powe
torchelie/data_learning.py:86
Method__init__
(self)
torchelie/data_learning.py:137
Method__init__
(self, *shape: int, init_sd: float = 0.06, init_img: torch.
torchelie/data_learning.py:196
Method__init__
(self, optimizer, schedule: List[Tuple[float, float, float]],
torchelie/lr_scheduler.py:23
Method__init__
(self, optimizer, total_iters: int, warmup_ratio: float = 0
torchelie/lr_scheduler.py:81
Method__init__
(self, optimizer, total_iters: int, warmup_ratio: float = 0
torchelie/lr_scheduler.py:108
Method__init__
(self, opt, lr: Tuple[float, float], num_iters: int,
torchelie/lr_scheduler.py:158
Method__init__
(self, optimizer: Optimizer, n_iters_total: int, tanh_lower
torchelie/lr_scheduler.py:241
Method__init__
(self, low: float, high: float)
torchelie/hyper.py:48
Method__init__
(self, low, high)
torchelie/hyper.py:97
Method__init__
(self, choices)
torchelie/hyper.py:118
Method__init__
(self, **hyperparams)
torchelie/hyper.py:153
Method__init__
(self, x, y)
torchelie/hyper.py:182
Method__init__
(self, params)
torchelie/hyper.py:237
Method__init__
(self, **hyperparams)
torchelie/hyper.py:259
Method__init__
(self, loc: torch.Tensor, scale: torch.Tensor)
torchelie/distributions.py:19
Method__init__
(self, weights, locs, scales, dim)
torchelie/distributions.py:43
Method__init__
(self, weights: torch.Tensor, locs: torch.Tensor, scales: torch.Tensor)
torchelie/distributions.py:74
Method__init__
(self, topk: int, labels: List[str], center_value: float =
torchelie/callbacks/inspector.py:60
Method__init__
(self, topk, labels, center_value=0)
torchelie/callbacks/inspector.py:155
Method__init__
(self)
torchelie/callbacks/avg.py:15
Method__init__
(self, beta: float = 0.6)
torchelie/callbacks/avg.py:78
Method__init__
(self, name, post_each_batch=True, k=100)
torchelie/callbacks/callbacks.py:44
Method__init__
(self, name, beta: float = 0.9, post_each_batch: bool = True)
torchelie/callbacks/callbacks.py:83
Method__init__
(self, name, post_each_batch=True)
torchelie/callbacks/callbacks.py:116
Method__init__
(self, k: int = 5, post_each_batch: bool = True, avg_type:
torchelie/callbacks/callbacks.py:155
Method__init__
(self, post_each_batch=True, avg_type='window')
torchelie/callbacks/callbacks.py:209
Method__init__
(self, post_each_batch=True, epoch_ends=True)
torchelie/callbacks/callbacks.py:226
Method__init__
(self, opt, accumulation=1, clip_grad_norm=None,
torchelie/callbacks/callbacks.py:293
Method__init__
(self, sched, metric='loss', step_each_batch=False)
torchelie/callbacks/callbacks.py:374
Method__init__
(self, visdom_env='main', log_every=10, prefix='',
torchelie/callbacks/callbacks.py:453
Method__init__
(self, log_dir=MISSING, log_every=10, prefix='',
torchelie/callbacks/callbacks.py:554
Method__init__
(self, log_every=10, prefix='')
torchelie/callbacks/callbacks.py:625
Method__init__
(self, filename_base, objects, max_saves=10, key_best=None)
torchelie/callbacks/callbacks.py:708
Method__init__
(self, original: nn.Module, copy: nn.Module, beta: float =
torchelie/callbacks/callbacks.py:776
Method__init__
(self)
torchelie/callbacks/callbacks.py:805
Method__init__
(self, nb_show, classes, post_each_batch=True)
torchelie/callbacks/callbacks.py:841
Method__init__
(self, nb_show, classes, post_each_batch=True)
torchelie/callbacks/callbacks.py:875
Method__init__
(self, labels, normalize=False)
torchelie/callbacks/callbacks.py:904
Method__init__
(self, loop)
torchelie/callbacks/callbacks.py:958
Method__init__
(self, loop, run_every=100, prefix='test', init_fun=None)
torchelie/callbacks/callbacks.py:980
Method__init__
(self)
torchelie/callbacks/callbacks.py:1019
Method__init__
( self, real_key: str = 'batch.0', fake_key: str = 'fake', dev
torchelie/callbacks/callbacks.py:1087
Method__init__
(self, real_key: str, fake_key: str, model: str = 'alex')
torchelie/callbacks/callbacks.py:1243
Method__init__
(self, n_scales=3)
torchelie/recipes/unpaired.py:15
Method__init__
(self, gamma)
torchelie/recipes/unpaired.py:226
Method__init__
(self, model, layer, input_size, *,
torchelie/recipes/feature_vis.py:38
Method__init__
(self, gamma)
torchelie/recipes/cut.py:24
Method__init__
(self, n_scales=3)
torchelie/recipes/cut.py:67
Method__init__
(self, gamma)
torchelie/recipes/cut.py:253
Method__init__
(self, device="cpu", visdom_env='style')
torchelie/recipes/neural_style.py:30
Method__init__
(self)
torchelie/recipes/algorithm.py:11
Method__init__
(self, target_loss: float, growth: float = 0.01)
torchelie/recipes/stylegan2.py:20
Method__init__
(self, every=4)
torchelie/recipes/stylegan2.py:61
Method__init__
(self, gamma)
torchelie/recipes/stylegan2.py:98
Method__init__
(self, gamma)
torchelie/recipes/pix2pix.py:97
Method__init__
(self, G: nn.Module, D: nn.Module)
torchelie/recipes/pix2pix.py:120
Method__init__
(self, G: nn.Module, D: nn.Module, l1_gain: float)
torchelie/recipes/pix2pix.py:229
Method__init__
(self, model, dream_layer)
torchelie/recipes/deepdream.py:41
Method__init__
(self)
torchelie/recipes/recipebase.py:8
Method__init__
(self)
torchelie/recipes/recipebase.py:85
Method__init__
(self, call_fun, loader)
torchelie/recipes/recipebase.py:231
Method__init__
(self, data)
torchelie/recipes/gan.py:51
Method__init__
(self, stride: int)
torchelie/transforms/differentiable.py:125
Method__init__
(self, B: int, init: Optional[torch.Tensor] = None)
torchelie/transforms/differentiable.py:183
Method__init__
(self, size, mode=Image.BILINEAR)
torchelie/transforms/__init__.py:21
Method__init__
(self, padding_mode='reflect', fill=0)
torchelie/transforms/__init__.py:44
Method__init__
(self, sz, padding_mode='constant', fill=0)
torchelie/transforms/__init__.py:71
Method__init__
(self, transforms)
torchelie/transforms/__init__.py:107
Method__init__
(self, size, scale=1, ratio=1, interpolati
torchelie/transforms/__init__.py:134
Method__init__
(self, std)
torchelie/transforms/__init__.py:203
Method__init__
(self, min_bits: int = 4, max_bits: int = 8)
torchelie/transforms/augments.py:44
Method__init__
(self, max_thresh: int = 128)
torchelie/transforms/augments.py:63
Method__init__
(self, min_size: float, max_size: float)
torchelie/transforms/augments.py:83
Method__init__
(self, min_ratio: float, max_ratio: float, patch_size: Opti
torchelie/transforms/augments.py:119
Method__init__
( self, min_ratio: int = 1, max_ratio: int = 3, p: float = 0.5
torchelie/transforms/augments.py:173
← previousnext →501–600 of 1,230, ranked by callers