MCPcopy Create free account

hub / github.com/Vermeille/Torchelie / functions

Functions1,230 in github.com/Vermeille/Torchelie

↓ 1 callersFunctiontrain
(model, loader)
examples/pixelcnn.py:14
↓ 1 callersFunctiontrain
()
examples/cifar.py:60
↓ 1 callersFunctiontrain_net
(Gen, Discr)
examples/gan.py:34
↓ 1 callersFunctiontrain_net
()
examples/conditional.py:63
↓ 1 callersMethodtranslate
Randomly translate image horizontally with an offset sampled in [-x, x] and vertically [-y, y]. Note that the coordinate are not pixe
torchelie/transforms/differentiable.py:195
↓ 1 callersFunctiontruncated_normal
(uniform: torch.Tensor, a: float, b: float)
torchelie/distributions.py:114
↓ 1 callersFunctiontup
(x)
torchelie/nn/graph.py:6
↓ 1 callersMethodunfolded_posenc
(self)
torchelie/nn/transformer.py:53
↓ 1 callersFunctionunfreeze
Unfreeze all parameters of `net`
torchelie/utils.py:37
↓ 1 callersMethodupdate_usage
(self, indices)
torchelie/nn/vq.py:56
↓ 1 callersMethoduse_r0
(self, r0_gamma)
torchelie/recipes/pix2pix.py:200
↓ 1 callersMethoduse_se
(self)
torchelie/nn/resblock.py:226
↓ 1 callersFunctionvgg13
(num_classes: int)
torchelie/models/vgg.py:72
↓ 1 callersFunctionvgg16
(num_classes: int)
torchelie/models/vgg.py:79
↓ 1 callersFunctionweight_norm_and_equal_lr
Set weight norm and equalized learning rate like demodulated convs in StyleGAN2 for module m. The weight matrix is initialized for a lea
torchelie/nn/utils.py:215
↓ 1 callersFunctionwith_patches
(img, patch_size, task, *args, **kwargs)
torchelie/recipes/image_prior.py:18
↓ 1 callersFunctionwith_patches2
(img, mask, patch_size, task, *args, **kwargs)
torchelie/recipes/image_prior.py:28
MethodD_fake
(env, src, dst)
torchelie/recipes/pix2pix.py:157
FunctionD_fun
(img)
wip/singan.py:122
FunctionD_fun
(batch)
torchelie/recipes/unpaired.py:248
FunctionD_fun
(batch)
torchelie/recipes/pix2pix.py:349
FunctionD_fun
(batch)
examples/gan.py:54
MethodD_r0
(env, src, dst)
torchelie/recipes/pix2pix.py:205
FunctionD_train
(batch)
torchelie/recipes/stylegan2.py:192
FunctionD_wrap
(batch)
torchelie/recipes/gan.py:24
MethodG_adv
(env, src, dst)
torchelie/recipes/pix2pix.py:260
MethodG_adv_pass
(env, src, dst)
torchelie/recipes/pix2pix.py:128
MethodG_featmatch
(env, src, dst)
torchelie/recipes/pix2pix.py:251
MethodG_features
(env, src, dst)
torchelie/recipes/pix2pix.py:237
FunctionG_fun
(img)
wip/singan.py:110
FunctionG_fun
(batch)
torchelie/recipes/unpaired.py:210
FunctionG_fun
(batch)
torchelie/recipes/pix2pix.py:343
FunctionG_fun
(batch)
examples/gan.py:41
MethodG_l1_pass
(env, src, dst)
torchelie/recipes/pix2pix.py:220
FunctionG_polyak_fun
(batch)
examples/gan.py:49
FunctionG_test
(state)
torchelie/recipes/gan.py:82
FunctionG_train
(batch)
torchelie/recipes/stylegan2.py:172
FunctionG_wrap
(batch)
torchelie/recipes/gan.py:30
FunctionR1
R1 regularizer from Which Training Methods for GANs do actually Converge? ( https://arxiv.org/abs/1801.04406 ). It forces grad norm of 0
torchelie/loss/gan/penalty.py:59
FunctionR2
R2 regularizer from Which Training Methods for GANs do actually Converge? ( https://arxiv.org/abs/1801.04406 ). It forces grad norm of 0
torchelie/loss/gan/penalty.py:83
FunctionR3
R2 regularizer from The GAN is dead; long live the GAN! A Modern Baseline GAN ( https://openreview.net/pdf?id=OrtN9hPP7V ). It forces gr
torchelie/loss/gan/penalty.py:107
FunctionStratifiedSampler
(dataset, num_samples=None)
torchelie/datasets/__init__.py:426
Method__call__
(self, x)
wip/adain.py:127
Method__call__
(self, images, klass: List[int])
torchelie/serving_utils.py:23
Method__call__
(self, *args, **kwargs)
torchelie/utils.py:455
Method__call__
(self, rank)
torchelie/utils.py:629
Method__call__
(self, img: torch.Tensor)
torchelie/data_learning.py:22
Method__call__
Correlate the color of the image `img` and return the result
torchelie/data_learning.py:149
Method__call__
(self, x: torch.Tensor)
torchelie/data_learning.py:171
Method__call__
(self, state)
torchelie/callbacks/callbacks.py:961
Method__call__
(self, model, real, fake)
torchelie/recipes/unpaired.py:231
Method__call__
(self, model, fake_dst, src, real_dst)
torchelie/recipes/cut.py:49
Method__call__
(self, model, real, fake)
torchelie/recipes/cut.py:258
Method__call__
(self, *args, **kwargs)
torchelie/recipes/algorithm.py:42
Method__call__
(self, x: torch.Tensor)
torchelie/recipes/stylegan2.py:26
Method__call__
(self, model, z)
torchelie/recipes/stylegan2.py:85
Method__call__
(self, model, real, fake)
torchelie/recipes/stylegan2.py:103
Method__call__
(self, model, real, fake)
torchelie/recipes/pix2pix.py:102
Method__call__
(self, name, *args, **kwargs)
torchelie/recipes/recipebase.py:15
Method__call__
Args: x (PIL.Image): the image to transform Returns: resized image
torchelie/transforms/__init__.py:25
Method__call__
(self, img)
torchelie/transforms/__init__.py:48
Method__call__
Pad the image Args: img (PIL.Image): The image to pad Returns: Padded image
torchelie/transforms/__init__.py:76
Method__call__
Transform the image Args: x: image Returns: A tuple `out` so that `out[i] = transforms[i](x)`
torchelie/transforms/__init__.py:110
Method__call__
Args: img (PIL Image): Image to be cropped and resized. Returns: PIL Image: Randomly cropped and resized imag
torchelie/transforms/__init__.py:184
Method__call__
(self, x)
torchelie/transforms/__init__.py:206
Method__call__
(self, x: PILImage)
torchelie/transforms/augments.py:48
Method__call__
(self, x: PILImage)
torchelie/transforms/augments.py:66
Method__call__
(self, x: PILImage)
torchelie/transforms/augments.py:87
Method__call__
(self, x: PILImage)
torchelie/transforms/augments.py:127
Method__call__
(self, x: PILImage)
torchelie/transforms/augments.py:156
Method__call__
(self, x: PILImage)
torchelie/transforms/augments.py:184
Method__call__
(self, x: PILImage)
torchelie/transforms/augments.py:218
Method__call__
(self, x: PILImage)
torchelie/transforms/augments.py:238
Method__call__
Detect edges Args: img (PIL.Image): the image Returns: edges detected in `img` as PIL Image
torchelie/transforms/augments.py:270
Method__call__
(self, x)
torchelie/transforms/augments.py:298
Method__call__
(self, x)
torchelie/transforms/augments.py:319
Method__call__
(self, img)
torchelie/transforms/augments.py:363
Method__call__
(self, module: Module, inputs: Any)
torchelie/nn/utils.py:75
Method__getattr__
(self, name)
torchelie/utils.py:438
Method__getattr__
(self, name)
torchelie/utils.py:458
Method__getattr__
(self, attr)
torchelie/datasets/__init__.py:157
Method__getitem__
(self, i: int)
wip/singan.py:86
Method__getitem__
(self, i)
wip/adain.py:112
Method__getitem__
(self, name: str)
torchelie/recipes/algorithm.py:86
Method__getitem__
(self, i)
torchelie/datasets/debug.py:32
Method__getitem__
(self, i)
torchelie/datasets/debug.py:57
Method__getitem__
(self, i)
torchelie/datasets/__init__.py:57
Method__getitem__
(self, i)
torchelie/datasets/__init__.py:95
Method__getitem__
(self, i)
torchelie/datasets/__init__.py:189
Method__getitem__
(self, i)
torchelie/datasets/__init__.py:212
Method__getitem__
(self, i)
torchelie/datasets/__init__.py:315
Method__getitem__
(self, i)
torchelie/datasets/__init__.py:339
Method__getitem__
Args: i (int): index Returns: A tuple (i, self.ds[i])
torchelie/datasets/__init__.py:366
Method__getitem__
Returns: The ith element of the underlying dataset or its cached value if available
torchelie/datasets/__init__.py:403
Method__getitem__
(self, i: int)
torchelie/datasets/pix2pix.py:35
Method__getitem__
(self, i: int)
torchelie/datasets/pix2pix.py:66
Method__getitem__
Return the ith paired image as (img1, img2)
torchelie/datasets/pix2pix.py:91
Method__getitem__
(self, i: int)
torchelie/datasets/ms1m.py:55
Method__getitem__
(self, i: int)
torchelie/datasets/concat.py:21
Method__getitem__
(self, i: int)
torchelie/datasets/concat.py:40
← previousnext →401–500 of 1,230, ranked by callers