MCPcopy Create free account

hub / github.com/ThisisBillhe/NAR / functions

Functions683 in github.com/ThisisBillhe/NAR

Method__init__
(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout, temb_channels=512)
NAR-images/tokenizer/vqgan/layer.py:58
Method__init__
(self, in_channels)
NAR-images/tokenizer/vqgan/layer.py:120
Method__init__
(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks, attn_resolutions, dropout=0.0, resam
NAR-images/tokenizer/vqgan/layer.py:270
Method__init__
(self, ddconfig, n_embed, embed_dim, ckpt_
NAR-images/tokenizer/vqgan/model.py:25
Method__init__
(self, directory, transform=None)
NAR-images/tokenizer/validation/val_ddp.py:18
Method__init__
(self, in_channels=3, ch=128, ch_mult=(1,1,2,2,4), num_res_blocks=2, norm_type='group', drop
NAR-images/tokenizer/tokenizer_image/vq_model.py:65
Method__init__
(self, z_channels=256, ch=128, ch_mult=(1,1,2,2,4), num_res_blocks=2, norm_type="group", drop
NAR-images/tokenizer/tokenizer_image/vq_model.py:129
Method__init__
(self, n_e, e_dim, beta, entropy_loss_ratio, l2_norm, show_usage)
NAR-images/tokenizer/tokenizer_image/vq_model.py:198
Method__init__
(self, in_channels, out_channels=None, conv_shortcut=False, dropout=0.0, norm_type='group')
NAR-images/tokenizer/tokenizer_image/vq_model.py:280
Method__init__
(self, in_channels, norm_type='group')
NAR-images/tokenizer/tokenizer_image/vq_model.py:318
Method__init__
(self, in_channels, with_conv)
NAR-images/tokenizer/tokenizer_image/vq_model.py:368
Method__init__
(self, in_channels, with_conv)
NAR-images/tokenizer/tokenizer_image/vq_model.py:382
Method__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
NAR-images/tokenizer/tokenizer_image/discriminator.py:21
Method__init__
(self, num_features, logdet=False, affine=True, allow_reverse_init=False)
NAR-images/tokenizer/tokenizer_image/discriminator.py:80
Method__init__
(self, input_nc=3, ndf=64, n_layers=3, channel_multiplier=1, image_size=256)
NAR-images/tokenizer/tokenizer_image/discriminator.py:169
Method__init__
(self, input_channels, filters, downsample=True)
NAR-images/tokenizer/tokenizer_image/discriminator.py:213
Method__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
NAR-images/tokenizer/tokenizer_image/discriminator_patchgan.py:12
Method__init__
(self, disc_start, disc_loss="hinge", disc_dim=64, disc_type='patchgan', image_size=256, disc
NAR-images/tokenizer/tokenizer_image/vq_loss.py:50
Method__init__
(self, input_nc=3, ndf=64, n_layers=3, channel_multiplier=1, image_size=256)
NAR-images/tokenizer/tokenizer_image/discriminator_stylegan.py:14
Method__init__
(self, input_channels, filters, downsample=True)
NAR-images/tokenizer/tokenizer_image/discriminator_stylegan.py:58
Method__init__
(self)
NAR-images/tokenizer/tokenizer_image/lpips.py:100
Method__init__
(self, chn_in, chn_out=1, use_dropout=False)
NAR-images/tokenizer/tokenizer_image/lpips.py:111
Method__init__
(self, requires_grad=False, pretrained=True)
NAR-images/tokenizer/tokenizer_image/lpips.py:119
Method__init__
(self, directory, transform=None)
NAR-images/tokenizer/consistencydecoder/reconstruction_cd_ddp.py:23
Method__init__
(self, num_classes, hidden_size, dropout_prob)
NAR-images/autoregressive/models/gpt.py:62
Method__init__
(self, in_channels, hidden_size, uncond_prob, token_num=120)
NAR-images/autoregressive/models/gpt.py:95
Method__init__
(self, dim: int, eps: float = 1e-5)
NAR-images/autoregressive/models/gpt.py:140
Method__init__
(self, config: ModelArgs)
NAR-images/autoregressive/models/gpt.py:154
Method__init__
(self, max_batch_size, max_seq_length, n_head, head_dim, dtype)
NAR-images/autoregressive/models/gpt.py:173
Method__init__
(self, config: ModelArgs)
NAR-images/autoregressive/models/gpt.py:191
Method__init__
(self, config: ModelArgs, drop_path: float)
NAR-images/autoregressive/models/gpt.py:247
Method__init__
(self, config: ModelArgs)
NAR-images/autoregressive/models/gpt.py:263
Method__init__
(self)
NAR-videos/utils/common.py:91
Method__init__
(self, max_epoch)
NAR-videos/utils/common.py:104
Method__init__
(self, data_folder: str, sequence_length: int = 16, resolution: int = 128, sample_every_n_frames: int = 1)
NAR-videos/utils/fvd/fvd.py:266
Method__init__
(self, i3d_path=None, device='cuda')
NAR-videos/utils/fvd/fvd.py:325
Method__init__
(self, data_folder: str, resolution: int = 256)
NAR-videos/utils/fid/fid.py:38
Method__init__
(self, dims=2048, device='cuda', version='stable', capture_all=False)
NAR-videos/utils/fid/fid.py:61
Method__init__
(self, in_channels, pool_features)
NAR-videos/utils/fid/inception.py:226
Method__init__
(self, in_channels, channels_7x7)
NAR-videos/utils/fid/inception.py:253
Method__init__
(self, in_channels)
NAR-videos/utils/fid/inception.py:283
Method__init__
(self, in_channels)
NAR-videos/utils/fid/inception.py:318
Method__init__
( self, model, dataset_csv, root_path='data/metadata', frame_num=16,
NAR-videos/eval/rfvd_evaluator.py:19
Method__init__
(self, rank, cfg)
NAR-videos/trainers/base_trainer.py:75
Method__init__
(self, rank, cfg)
NAR-videos/trainers/larp_tokenizer_trainer.py:34
Method__init__
(self, rank, cfg)
NAR-videos/trainers/larp_ar_fp_trainer.py:35
Method__init__
(self, rank, cfg)
NAR-videos/trainers/larp_ar_trainer.py:28
Method__init__
( self, root_path, frame_num, cls_vid_num, crop_size, rand_fli
NAR-videos/datasets/video_dataset.py:61
Method__init__
( self, disc_start, disc_self_start=None, pixelloss_weight=1.0, disc_t
NAR-videos/models/loss.py:209
Method__init__
(self, drop_prob: float = 0., scale_by_keep: bool = True)
NAR-videos/models/larp_ar.py:115
Method__init__
(self, config: ModelArgs)
NAR-videos/models/larp_ar.py:132
Method__init__
(self, max_batch_size, max_seq_length, n_head, head_dim, dtype)
NAR-videos/models/larp_ar.py:151
Method__init__
(self, config: ModelArgs)
NAR-videos/models/larp_ar.py:169
Method__init__
(self, config: ModelArgs, drop_path: float)
NAR-videos/models/larp_ar.py:222
Method__init__
(self, config: ModelArgs)
NAR-videos/models/larp_ar.py:238
Method__init__
( self, bottleneck, prior_model, bottleneck_token_num=1024, input_siz
NAR-videos/models/larp_tokenizer.py:46
Method__init__
(self, dim: int, eps: float = 1e-5)
NAR-videos/models/norm.py:7
Method__init__
( self, dim, depth, n_head, head_dim, ff_dim=None, dro
NAR-videos/models/transformer.py:36
Method__init__
(self, *args, **kwargs)
NAR-videos/models/embed.py:17
Method__init__
(self, hidden_size, frequency_embedding_size=256)
NAR-videos/models/embed.py:127
Method__init__
(self, codebook_size, hidden_size, dropout_prob)
NAR-videos/models/embed.py:171
Method__init__
(self, token_dim, hidden_szie, dropout_prob)
NAR-videos/models/embed.py:203
Method__init__
(self, num_classes, hidden_size, dropout_prob)
NAR-videos/models/embed.py:233
Method__init__
(self, parameters, deterministic=True)
NAR-videos/models/bottleneck.py:37
Method__init__
(self, *args, **kwargs)
NAR-videos/models/bottleneck.py:193
Method__init__
( self, dim, codebook_size, commitment_loss_weight=0.25, entropy_loss_
NAR-videos/models/bottleneck.py:205
Method__init__
( self, dim, **kwargs, )
NAR-videos/models/bottleneck.py:349
Method__init__
(self, config)
NAR-videos/models/gptc.py:37
Method__init__
(self, config)
NAR-videos/models/gptc.py:81
Method__iter__
(self)
NAR-images/evaluations/c2i/evaluator.py:479
Method__len__
(self)
NAR-images/evaluations/c2i/evaluator.py:476
Method__len__
(self)
NAR-images/evaluations/t2i/evaluation.py:100
Method__len__
(self)
NAR-images/dataset/imagenet.py:29
Method__len__
(self)
NAR-images/dataset/t2i.py:39
Method__len__
(self)
NAR-images/dataset/t2i.py:85
Method__len__
(self)
NAR-images/dataset/openimage.py:20
Method__len__
(self)
NAR-images/dataset/coco.py:15
Method__len__
(self)
NAR-images/language/extract_t5_feature.py:41
Method__len__
(self)
NAR-images/tokenizer/vae/reconstruction_vae_ddp.py:30
Method__len__
(self)
NAR-images/tokenizer/vqgan/reconstruction_vqgan_ddp.py:32
Method__len__
(self)
NAR-images/tokenizer/validation/val_ddp.py:25
Method__len__
(self)
NAR-images/tokenizer/consistencydecoder/reconstruction_cd_ddp.py:30
Method__len__
(self)
NAR-videos/utils/fvd/fvd.py:307
Method__len__
(self)
NAR-videos/utils/fid/fid.py:51
Method__len__
(self)
NAR-videos/datasets/video_dataset.py:307
Method__repr__
(self)
NAR-images/evaluations/t2i/evaluation.py:40
Method_basic_init
(module)
NAR-videos/models/loss.py:169
Method_basic_init
(module)
NAR-videos/models/larp_tokenizer.py:245
Method_init_weights
(self, module)
NAR-images/tokenizer/tokenizer_image/discriminator.py:67
Method_init_weights
(self, module)
NAR-images/tokenizer/tokenizer_image/discriminator_patchgan.py:58
Method_init_weights
(self, module)
NAR-images/autoregressive/models/gpt.py:315
Method_init_weights
(self, module)
NAR-videos/models/larp_ar.py:302
Method_init_weights
(self, module)
NAR-videos/models/gptc.py:131
Functionar_make
(model_spec, args=None, load_sd=False)
NAR-videos/models/models.py:29
Methodbest_psnr
(psnr_list, precision=2)
NAR-videos/trainers/base_trainer.py:467
Methodcalculate_fid_original
( self, feats_gen: FeatureStats, feats_real: FeatureStats, eps=1e-6, )
NAR-videos/utils/fid/fid.py:116
Methodcalculate_fid_stable
( self, feats_gen: FeatureStats, feats_real: FeatureStats, )
NAR-videos/utils/fid/fid.py:163
Methodcalculate_fvd_with_video_folder
( self, feats_real, video_folder, bs=32, num_workers=4, seq
NAR-videos/utils/fvd/fvd.py:440
Functioncalculate_md5
Calculate the MD5 checksum of a file.
NAR-images/scripts/analyze_tar.py:6
Functioncalculate_topk_accuracy
Computes the top-k accuracy for the specified values of k. Args: logits (torch.Tensor): The predicted logits (unnormalized scores) with
NAR-videos/utils/statistics.py:36
← previousnext →401–500 of 683, ranked by callers