MCPcopy Create free account

hub / github.com/Kwai-Klear/AR-GRPO / functions

Functions650 in github.com/Kwai-Klear/AR-GRPO

⨍ Functions 650◇ Types & classes 121↳ Endpoints 2
Method__init__
(self,**kwargs)
img_gen_grpo_rewards.py:595
Method__init__
(self, mu: np.ndarray, sigma: np.ndarray)
fid_evaluation.py:72
Method__init__
( self, session, batch_size=64, softmax_batch_size=512, )
fid_evaluation.py:123
Method__init__
Estimate the manifold of given feature vectors. :param session: the TensorFlow session. :param row_batch_size: row batch siz
fid_evaluation.py:216
Method__init__
(self, session)
fid_evaluation.py:373
Method__init__
(self, gen_fn, length)
fid_evaluation.py:460
Method__init__
(self, arr_f, shape, dtype)
fid_evaluation.py:472
Method__init__
(self, arr)
fid_evaluation.py:498
Method__init__
(self, directory, transform=None)
tokenizer_lg/vae/reconstruction_vae_ddp.py:23
Method__init__
(self, directory, transform=None)
tokenizer_lg/vqgan/reconstruction_vqgan_ddp.py:25
Method__init__
(self, n_e, e_dim, beta, remap=None, unknown_index="random", sane_index_shape=False, legacy=T
tokenizer_lg/vqgan/quantize.py:118
Method__init__
(self, in_channels, with_conv)
tokenizer_lg/vqgan/layer.py:18
Method__init__
(self, in_channels, with_conv)
tokenizer_lg/vqgan/layer.py:36
Method__init__
(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout, temb_channels=512)
tokenizer_lg/vqgan/layer.py:58
Method__init__
(self, in_channels)
tokenizer_lg/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
tokenizer_lg/vqgan/layer.py:270
Method__init__
(self, ddconfig, n_embed, embed_dim, ckpt_
tokenizer_lg/vqgan/model.py:25
Method__init__
(self, directory, transform=None)
tokenizer_lg/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
tokenizer_lg/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
tokenizer_lg/tokenizer_image/vq_model.py:129
Method__init__
(self, n_e, e_dim, beta, entropy_loss_ratio, l2_norm, show_usage)
tokenizer_lg/tokenizer_image/vq_model.py:198
Method__init__
(self, in_channels, out_channels=None, conv_shortcut=False, dropout=0.0, norm_type='group')
tokenizer_lg/tokenizer_image/vq_model.py:280
Method__init__
(self, in_channels, norm_type='group')
tokenizer_lg/tokenizer_image/vq_model.py:318
Method__init__
(self, in_channels, with_conv)
tokenizer_lg/tokenizer_image/vq_model.py:368
Method__init__
(self, in_channels, with_conv)
tokenizer_lg/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)
tokenizer_lg/tokenizer_image/discriminator.py:21
Method__init__
(self, num_features, logdet=False, affine=True, allow_reverse_init=False)
tokenizer_lg/tokenizer_image/discriminator.py:80
Method__init__
(self, input_nc=3, ndf=64, n_layers=3, channel_multiplier=1, image_size=256)
tokenizer_lg/tokenizer_image/discriminator.py:169
Method__init__
(self, input_channels, filters, downsample=True)
tokenizer_lg/tokenizer_image/discriminator.py:213
Method__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
tokenizer_lg/tokenizer_image/discriminator_patchgan.py:12
Method__init__
(self, disc_start, disc_loss="hinge", disc_dim=64, disc_type='patchgan', image_size=256, disc
tokenizer_lg/tokenizer_image/vq_loss.py:50
Method__init__
(self, input_nc=3, ndf=64, n_layers=3, channel_multiplier=1, image_size=256)
tokenizer_lg/tokenizer_image/discriminator_stylegan.py:14
Method__init__
(self, input_channels, filters, downsample=True)
tokenizer_lg/tokenizer_image/discriminator_stylegan.py:58
Method__init__
(self)
tokenizer_lg/tokenizer_image/lpips.py:100
Method__init__
(self, chn_in, chn_out=1, use_dropout=False)
tokenizer_lg/tokenizer_image/lpips.py:111
Method__init__
(self, requires_grad=False, pretrained=True)
tokenizer_lg/tokenizer_image/lpips.py:119
Method__init__
(self, directory, transform=None)
tokenizer_lg/consistencydecoder/reconstruction_cd_ddp.py:23
Method__init__
(self, drop_prob: float = 0., scale_by_keep: bool = True)
utils/drop_path.py:27
Method__init__
(self, lst_dir, start, end, caption_key, trunc_caption=False)
language/extract_t5_feature.py:24
Method__init__
(self, device, dir_or_name='t5-v1_1-xxl', *, local_cache=False, cache_dir=None, hf_token=None, use_text_prepro
language/t5.py:20
Method__init__
(self, image: Image.Image, objects)
benchmark/reward-server/reward_server/gen_eval.py:65
Method__init__
(self, image: Image.Image, objects)
benchmark/geneval/evaluation/evaluate_images.py:83
Method__init__
(self, image_path, transform, num_crops=20)
reward_utils/maniqa.py:12
Method__init__
(self, device="cuda", dtype=torch.float32)
reward_utils/imagereward.py:7
Method__init__
(self, files, transforms=None)
reward_utils/fid.py:50
Method__init__
(self, device="cuda", dtype=torch.float32)
reward_utils/pickscore.py:12
Method__init__
(self, dtype)
reward_utils/aesthscore.py:37
Method__init__
(self, text_encoder=None, image_tokenizer=None, codebook_embed_dim=None, **kwargs)
autoregressive/modeling.py:26
Method__init__
( self, **kwargs, )
autoregressive/modeling.py:102
Method__init__
( self, args: argparse.ArgumentParser, model: str, tokenizer: Optional[str] =
autoregressive/serve/llm.py:82
Method__init__
( self, model_config: ModelConfig, parallel_config: ParallelConfig, scheduler_
autoregressive/serve/model_runner.py:105
Method__init__
(self, model: nn.Module)
autoregressive/serve/model_runner.py:1100
Method__init__
( self, args: argparse.ArgumentParser, model_config: ModelConfig, cache_config
autoregressive/serve/gpu_executor.py:18
Method__init__
(self, num_classes, hidden_size, dropout_prob)
autoregressive/serve/gpt_model.py:54
Method__init__
(self, config: ModelArgs)
autoregressive/serve/gpt_model.py:99
Method__init__
(self, config: ModelArgs)
autoregressive/serve/gpt_model.py:185
Method__init__
(self, config: ModelArgs)
autoregressive/serve/gpt_model.py:228
Method__init__
(self, config: ModelArgs)
autoregressive/serve/gpt_model.py:245
Method__init__
(self, cfg_scale=1.0)
autoregressive/serve/sampler.py:38
Method__init__
( self, args: argparse.ArgumentParser, model_config: ModelConfig, cache_config
autoregressive/serve/llm_engine.py:85
Method__init__
( self, model_config: ModelConfig, parallel_config: ParallelConfig, scheduler_
autoregressive/serve/worker.py:35
Method__init__
(self, num_classes, hidden_size, dropout_prob)
autoregressive/models/gpt.py:60
Method__init__
(self, in_channels, hidden_size, uncond_prob, token_num=120)
autoregressive/models/gpt.py:93
Method__init__
(self, dim: int, eps: float = 1e-5)
autoregressive/models/gpt.py:138
Method__init__
(self, config: ModelArgs)
autoregressive/models/gpt.py:152
Method__init__
(self, max_batch_size, max_seq_length, n_head, head_dim, dtype)
autoregressive/models/gpt.py:171
Method__init__
(self, config: ModelArgs)
autoregressive/models/gpt.py:189
Method__init__
(self, config: ModelArgs, drop_path: float)
autoregressive/models/gpt.py:245
Method__init__
(self, config: ModelArgs)
autoregressive/models/gpt.py:261
Method__init__
(self, lst_dir, start, end, transform)
autoregressive/train/extract_codes_t2i.py:26
Method__iter__
(self)
modified_grpo_trainer.py:158
Method__iter__
(self)
fid_evaluation.py:467
Method__len__
(self)
modified_grpo_trainer.py:179
Method__len__
(self)
lazy_dataset.py:67
Method__len__
(self)
lazy_dataset.py:165
Method__len__
(self)
fid_evaluation.py:464
Method__len__
(self)
tokenizer_lg/vae/reconstruction_vae_ddp.py:30
Method__len__
(self)
tokenizer_lg/vqgan/reconstruction_vqgan_ddp.py:32
Method__len__
(self)
tokenizer_lg/validation/val_ddp.py:25
Method__len__
(self)
tokenizer_lg/consistencydecoder/reconstruction_cd_ddp.py:30
Method__len__
(self)
language/extract_t5_feature.py:41
Method__len__
(self)
benchmark/reward-server/reward_server/gen_eval.py:74
Method__len__
(self)
benchmark/geneval/evaluation/evaluate_images.py:92
Method__len__
(self)
reward_utils/fid.py:54
Method__len__
(self)
autoregressive/train/extract_codes_t2i.py:41
Method__reduce__
(self)
autoregressive/serve/llm_engine.py:293
Method_get_eval_sampler
(self, eval_dataset)
modified_grpo_trainer.py:820
Method_init_weights
(self, module)
tokenizer_lg/tokenizer_image/discriminator.py:67
Method_init_weights
(self, module)
tokenizer_lg/tokenizer_image/discriminator_patchgan.py:58
Method_init_weights
(self, module)
autoregressive/models/gpt.py:307
Method_reward_func
(self, cont, lbl, **kwargs)
img_gen_grpo_rewards.py:200
Method_reward_func
(self, cont, lbl, **kwargs)
img_gen_grpo_rewards.py:234
Method_reward_func
(self,cnt,lbl,**kwargs)
img_gen_grpo_rewards.py:278
Method_reward_func
(self,cnt,lbl,**kwargs)
img_gen_grpo_rewards.py:323
Method_reward_func
(self,cnt,lbl,**kwargs)
img_gen_grpo_rewards.py:351
Method_reward_func
(self,cnt,lbl,**kwargs)
img_gen_grpo_rewards.py:377
Method_reward_func
(self,cnt,sol,**kwargs)
img_gen_grpo_rewards.py:464
Method_reward_func
(self,cnt,sol,**kwargs)
img_gen_grpo_rewards.py:478
Method_reward_func
(self, cont, sol, **kwargs)
img_gen_grpo_rewards.py:503
Method_reward_func
(self,cnt,**kwargs)
img_gen_grpo_rewards.py:550
← previousnext →401–500 of 650, ranked by callers