Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Vermeille/Torchelie
/ functions
Functions
1,230 in github.com/Vermeille/Torchelie
⨍
Functions
1,230
◇
Types & classes
261
↓ 1 callers
Function
train
(model, loader)
examples/pixelcnn.py:14
↓ 1 callers
Function
train
()
examples/cifar.py:60
↓ 1 callers
Function
train_net
(Gen, Discr)
examples/gan.py:34
↓ 1 callers
Function
train_net
()
examples/conditional.py:63
↓ 1 callers
Method
translate
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 callers
Function
truncated_normal
(uniform: torch.Tensor, a: float, b: float)
torchelie/distributions.py:114
↓ 1 callers
Function
tup
(x)
torchelie/nn/graph.py:6
↓ 1 callers
Method
unfolded_posenc
(self)
torchelie/nn/transformer.py:53
↓ 1 callers
Function
unfreeze
Unfreeze all parameters of `net`
torchelie/utils.py:37
↓ 1 callers
Method
update_usage
(self, indices)
torchelie/nn/vq.py:56
↓ 1 callers
Method
use_r0
(self, r0_gamma)
torchelie/recipes/pix2pix.py:200
↓ 1 callers
Method
use_se
(self)
torchelie/nn/resblock.py:226
↓ 1 callers
Function
vgg13
(num_classes: int)
torchelie/models/vgg.py:72
↓ 1 callers
Function
vgg16
(num_classes: int)
torchelie/models/vgg.py:79
↓ 1 callers
Function
weight_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 callers
Function
with_patches
(img, patch_size, task, *args, **kwargs)
torchelie/recipes/image_prior.py:18
↓ 1 callers
Function
with_patches2
(img, mask, patch_size, task, *args, **kwargs)
torchelie/recipes/image_prior.py:28
Method
D_fake
(env, src, dst)
torchelie/recipes/pix2pix.py:157
Function
D_fun
(img)
wip/singan.py:122
Function
D_fun
(batch)
torchelie/recipes/unpaired.py:248
Function
D_fun
(batch)
torchelie/recipes/pix2pix.py:349
Function
D_fun
(batch)
examples/gan.py:54
Method
D_r0
(env, src, dst)
torchelie/recipes/pix2pix.py:205
Function
D_train
(batch)
torchelie/recipes/stylegan2.py:192
Function
D_wrap
(batch)
torchelie/recipes/gan.py:24
Method
G_adv
(env, src, dst)
torchelie/recipes/pix2pix.py:260
Method
G_adv_pass
(env, src, dst)
torchelie/recipes/pix2pix.py:128
Method
G_featmatch
(env, src, dst)
torchelie/recipes/pix2pix.py:251
Method
G_features
(env, src, dst)
torchelie/recipes/pix2pix.py:237
Function
G_fun
(img)
wip/singan.py:110
Function
G_fun
(batch)
torchelie/recipes/unpaired.py:210
Function
G_fun
(batch)
torchelie/recipes/pix2pix.py:343
Function
G_fun
(batch)
examples/gan.py:41
Method
G_l1_pass
(env, src, dst)
torchelie/recipes/pix2pix.py:220
Function
G_polyak_fun
(batch)
examples/gan.py:49
Function
G_test
(state)
torchelie/recipes/gan.py:82
Function
G_train
(batch)
torchelie/recipes/stylegan2.py:172
Function
G_wrap
(batch)
torchelie/recipes/gan.py:30
Function
R1
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
Function
R2
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
Function
R3
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
Function
StratifiedSampler
(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
← previous
next →
401–500 of 1,230, ranked by callers