MCPcopy Create free account

hub / github.com/RameenAbdal/CLIP2StyleGAN / types & classes

Types & classes64 in github.com/RameenAbdal/CLIP2StyleGAN

↓ 9 callersClassConvLayer
edits_sg/model.py:573
↓ 7 callersClassEqualLinear
edits_sg/model.py:132
↓ 7 callersClassNetLinLayer
A single linear layer which does a 1x1 conv
edits_sg/lpips/networks_basic.py:104
↓ 6 callersClassGenerator
edits_sg/model.py:391
↓ 6 callersClassStyledConv
edits_sg/model.py:332
↓ 5 callersClassLayerNorm
Subclass torch's LayerNorm to handle fp16.
clipmod/model.py:153
↓ 4 callersClassFIDInceptionC
InceptionC block patched for FID computation
edits_sg/inception.py:218
↓ 4 callersClassGenerator
edits_sg/swagan.py:145
↓ 3 callersClassBlur
edits_sg/model.py:75
↓ 3 callersClassFIDInceptionA
InceptionA block patched for FID computation
edits_sg/inception.py:193
↓ 3 callersClassHaarTransform
edits_sg/swagan.py:78
↓ 3 callersClassInverseHaarTransform
edits_sg/swagan.py:98
↓ 3 callersClassModulatedConv2d
edits_sg/model.py:169
↓ 2 callersClassBottleneck
clipmod/model.py:10
↓ 2 callersClassConstantInput
edits_sg/model.py:319
↓ 2 callersClassFromRGB
edits_sg/swagan.py:343
↓ 2 callersClassFusedLeakyReLU
edits_sg/op/fused_act.py:93
↓ 2 callersClassPixelNorm
edits_sg/model.py:14
↓ 2 callersClassToRGB
edits_sg/swagan.py:119
↓ 2 callersClassToRGB
edits_sg/model.py:369
↓ 2 callersClassTransformer
clipmod/model.py:191
↓ 2 callersClassUpsample
edits_sg/model.py:33
↓ 1 callersClassAttentionPool2d
clipmod/model.py:56
↓ 1 callersClassCLIP
clipmod/model.py:239
↓ 1 callersClassConvBlock
edits_sg/swagan.py:329
↓ 1 callersClassDiscriminator
edits_sg/model.py:639
↓ 1 callersClassDist2LogitLayer
takes 2 distances, puts through fc layers, spits out value between [0,1] (if use_sigmoid is True)
edits_sg/lpips/networks_basic.py:114
↓ 1 callersClassDownsample
edits_sg/model.py:54
↓ 1 callersClassEqualConv2d
edits_sg/model.py:94
↓ 1 callersClassFIDInceptionE_1
First InceptionE block patched for FID computation
edits_sg/inception.py:246
↓ 1 callersClassFIDInceptionE_2
Second InceptionE block patched for FID computation
edits_sg/inception.py:279
↓ 1 callersClassInceptionV3
Pretrained InceptionV3 network returning feature maps
edits_sg/inception.py:16
↓ 1 callersClassModifiedResNet
A ResNet class that is similar to torchvision's but contains the following changes: - There are now 3 "stem" convolutions as opposed to 1, wi
clipmod/model.py:93
↓ 1 callersClassMultiResolutionDataset
edits_sg/dataset.py:8
↓ 1 callersClassNoiseInjection
edits_sg/model.py:305
↓ 1 callersClassQuickGELU
clipmod/model.py:162
↓ 1 callersClassResBlock
edits_sg/model.py:618
↓ 1 callersClassResidualAttentionBlock
clipmod/model.py:167
↓ 1 callersClassScalingLayer
edits_sg/lpips/networks_basic.py:94
↓ 1 callersClassVisionTransformer
clipmod/model.py:202
ClassAdaptiveAugment
edits_sg/non_leaking.py:12
ClassBCERankingLoss
edits_sg/lpips/networks_basic.py:131
ClassBaseModel
edits_sg/lpips/base_model.py:8
ClassConv2d
edits_sg/op/conv2d_gradfix.py:134
ClassConv2dGradWeight
edits_sg/op/conv2d_gradfix.py:177
ClassDSSIM
edits_sg/lpips/networks_basic.py:167
ClassDiscriminator
edits_sg/swagan.py:370
ClassDistModel
edits_sg/lpips/dist_model.py:24
ClassFakeNet
edits_sg/lpips/networks_basic.py:144
ClassFusedLeakyReLUFunction
edits_sg/op/fused_act.py:62
ClassFusedLeakyReLUFunctionBackward
edits_sg/op/fused_act.py:20
ClassGridSampleBackward
edits_sg/non_leaking.py:357
ClassGridSampleForward
edits_sg/non_leaking.py:339
ClassInception3Feature
edits_sg/calc_inception.py:18
ClassL2
edits_sg/lpips/networks_basic.py:150
ClassPNetLin
edits_sg/lpips/networks_basic.py:27
ClassPerceptualLoss
edits_sg/lpips/__init__.py:13
ClassSimpleTokenizer
clipmod/simple_tokenizer.py:62
ClassUpFirDn2d
edits_sg/op/upfirdn2d.py:89
ClassUpFirDn2dBackward
edits_sg/op/upfirdn2d.py:20
Classalexnet
edits_sg/lpips/pretrained_networks.py:57
Classresnet
edits_sg/lpips/pretrained_networks.py:139
Classsqueezenet
edits_sg/lpips/pretrained_networks.py:6
Classvgg16
edits_sg/lpips/pretrained_networks.py:97