MCPcopy Create free account

hub / github.com/JialianW/GRiT / functions

Functions187 in github.com/JialianW/GRiT

⨍ Functions 187◇ Types & classes 41
Method__init__
( self, input_shape: ShapeSpec, **kwargs, )
grit/modeling/roi_heads/grit_fast_rcnn.py:19
Method__init__
( self, *, text_decoder_transformer, train_task: list, test_task: str,
grit/modeling/roi_heads/grit_roi_heads.py:29
Method__init__
( self, **kwargs)
grit/modeling/meta_arch/grit.py:12
Method__init__
(self, config)
grit/modeling/text/modeling_bert.py:71
Method__init__
(self, config)
grit/modeling/text/modeling_bert.py:137
Method__init__
(self, config)
grit/modeling/text/modeling_bert.py:156
Method__init__
(self, config)
grit/modeling/text/modeling_bert.py:178
Method__init__
(self, config)
grit/modeling/text/modeling_bert.py:191
Method__init__
(self, config, use_act_checkpoint=True)
grit/modeling/text/modeling_bert.py:228
Method__init__
(self, config, use_act_checkpoint=True)
grit/modeling/text/modeling_bert.py:264
Method__init__
(self, **kwargs)
grit/modeling/text/modeling_bert.py:308
Method__init__
(self, vocab_size_or_config_json_file=30522, hidden_size=768,
grit/modeling/text/modeling_bert.py:489
Method__init__
(self, tokenizer, max_text_len=40, padding='do_not_pad')
grit/modeling/text/load_text_token.py:5
Method__init__
( self, vocab_size: int, hidden_size: int, dropout: float = 0.0, max_c
grit/modeling/text/text_decoder.py:25
Method__init__
(self, encoder)
grit/modeling/text/text_decoder.py:75
Method__init__
( self, object_feature_size: int, vocab_size: int, hidden_size: int, n
grit/modeling/text/text_decoder.py:230
Method__init__
( self, end_token_id: int, max_steps: int = 50, beam_size: int = 5, ob
grit/modeling/text/text_decoder.py:357
Method__init__
( self, transformer, begin_token_id=101, beamsearch_decode=None, loss_
grit/modeling/text/text_decoder.py:543
Method__init__
(self, eps=0.1, log_prefix='', ignore_index=None)
grit/modeling/text/text_decoder.py:645
Method__init__
( self, dataset_dicts, dataset_ratio, seed: Optional[int] = None, )
grit/data/custom_dataset_dataloader.py:158
Method__init__
(self, dataset, batch_sizes)
grit/data/custom_dataset_dataloader.py:213
Method__init__
(self, dataset, batch_sizes, num_datasets)
grit/data/custom_dataset_dataloader.py:231
Method__init__
(self, is_train: bool, dataset_augs=[], **kwargs)
grit/data/custom_dataset_mapper.py:22
Method__init__
Args: h, w (int): original image size new_h, new_w (int): new image size interp: PIL interpolation method
grit/data/transforms/custom_transform.py:33
Method__init__
grit/data/transforms/custom_augmentation_impl.py:23
Method__iter__
(self)
grit/data/custom_dataset_dataloader.py:190
Method__iter__
(self)
grit/data/custom_dataset_dataloader.py:218
Method__iter__
(self)
grit/data/custom_dataset_dataloader.py:238
Method__len__
(self)
grit/data/custom_dataset_mapper.py:145
Method__repr__
(self)
grit/modeling/text/modeling_bert.py:440
Method__repr__
(self)
grit/data/custom_dataset_mapper.py:148
Function_custom_train_loader_from_config
(cfg, mapper=None, *, dataset=None, sampler=None)
grit/data/custom_dataset_dataloader.py:24
Function_distributed_worker
Modified by Jialian Wu from https://github.com/facebookresearch/detectron2/blob/main/detectron2/engine/launch.py Adaptation for deepspeed
lauch_deepspeed.py:85
Method_eval_predictions
(self, predictions, img_ids=None)
grit/evaluation/eval.py:24
Method_eval_predictions
This is only for saving the results to json file
grit/evaluation/eval.py:110
Function_gelu_python
(x)
grit/modeling/text/modeling_bert.py:527
Method_init_box_head
(self, cfg, input_shape)
grit/modeling/roi_heads/grit_roi_heads.py:107
Method_init_weights
(self, m)
grit/modeling/backbone/vit.py:341
Method_init_weights
Initialize weights like BERT - N(0.0, 0.02), bias = 0.
grit/modeling/text/text_decoder.py:289
Methodapply_coords
(self, coords)
grit/data/transforms/custom_transform.py:81
Methodapply_segmentation
(self, segmentation)
grit/data/transforms/custom_transform.py:89
Functionbuild_vit_fpn_backbone
(cfg, input_shape: ShapeSpec)
grit/modeling/backbone/vit.py:437
Functionbuild_vit_fpn_backbone_huge
(cfg, input_shape: ShapeSpec)
grit/modeling/backbone/vit.py:510
Functionbuild_vit_fpn_backbone_large
(cfg, input_shape: ShapeSpec)
grit/modeling/backbone/vit.py:478
Methoddecoding_step
(self, object_features, partial_text)
grit/modeling/text/text_decoder.py:621
Functionfilename_to_url
Return the url and etag (which may be ``None``) stored for `filename`. Raise ``EnvironmentError`` if `filename` or its stored metadata do not
grit/modeling/text/file_utils.py:66
Methodforward
(self, x)
grit/modeling/backbone/vit.py:72
Methodforward
(self, x)
grit/modeling/backbone/vit.py:143
Methodforward
(self, x)
grit/modeling/backbone/vit.py:215
Methodforward
(self, x)
grit/modeling/backbone/vit.py:350
Methodforward
(self, x)
grit/modeling/backbone/vit.py:395
Methodforward
(self, x)
grit/modeling/backbone/utils.py:182
Methodforward
(self, x)
grit/modeling/roi_heads/grit_fast_rcnn.py:116
Methodforward
(self, features, proposals, targets=None, targets_task="ObjectDet")
grit/modeling/roi_heads/grit_roi_heads.py:297
Methodforward
(self, batched_inputs: List[Dict[str, torch.Tensor]])
grit/modeling/meta_arch/grit.py:44
Methodforward
(self, query, key, attention_mask, gamma)
grit/modeling/text/modeling_bert.py:63
Methodforward
(self, hidden_states, attention_mask, head_mask=None, history_state=None)
grit/modeling/text/modeling_bert.py:100
Methodforward
(self, hidden_states, input_tensor)
grit/modeling/text/modeling_bert.py:145
Methodforward
(self, input_tensor, attention_mask, head_mask=None, history_state=None)
grit/modeling/text/modeling_bert.py:164
Methodforward
(self, hidden_states)
grit/modeling/text/modeling_bert.py:184
Methodforward
(self, hidden_states, input_tensor)
grit/modeling/text/modeling_bert.py:199
Methodforward
(self, attention_output)
grit/modeling/text/modeling_bert.py:218
Methodforward
(self, hidden_states, attention_mask, head_mask=None, history_state=None)
grit/modeling/text/modeling_bert.py:242
Methodforward
(self, hidden_states, attention_mask, head_mask=None, encoder_history_states=None)
grit/modeling/text/modeling_bert.py:273
Methodforward
(self, tokens: torch.Tensor)
grit/modeling/text/text_decoder.py:48
Methodforward
(self, tgt, memory, tgt_mask=None, tgt_key_padding_mask=None,
grit/modeling/text/text_decoder.py:79
Methodforward
(self, tgt, memory, tgt_mask=None, memory_mask=None, tgt_key_padding_mask=None, memory_key_pad
grit/modeling/text/text_decoder.py:202
Methodforward
( self, hidden_states, text_tokens, )
grit/modeling/text/text_decoder.py:302
Methodforward
(self, batch)
grit/modeling/text/text_decoder.py:566
Methodforward
(self, feature, target)
grit/modeling/text/text_decoder.py:657
Methodfrom_config
(cls, cfg, input_shape)
grit/modeling/roi_heads/grit_fast_rcnn.py:42
Methodfrom_config
(cls, cfg, input_shape)
grit/modeling/roi_heads/grit_roi_heads.py:81
Methodfrom_config
(cls, cfg)
grit/modeling/meta_arch/grit.py:19
Methodfrom_config
(cls, cfg, is_train: bool = True)
grit/data/custom_dataset_mapper.py:30
Methodget_transform
(self, img)
grit/data/transforms/custom_augmentation_impl.py:33
Methodinverse
(self)
grit/data/transforms/custom_transform.py:94
Methodinverse_apply_box
grit/data/transforms/custom_transform.py:106
Functionmain
(args)
train_deepspeed.py:234
Functionmain
(args)
train_net.py:192
Methodoutput_shape
(self)
grit/modeling/backbone/vit.py:427
Methodprocess
(self, inputs, outputs)
grit/evaluation/eval.py:13
Methodprocess
(self, inputs, outputs)
grit/evaluation/eval.py:87
Functions3_request
Wrapper function for s3 requests in order to create more helpful error messages.
grit/modeling/text/file_utils.py:135
Methodsave_pretrained
Save a configuration object to a directory, so that it can be re-loaded using the `from_pretrained(save_directory)` class method.
grit/modeling/text/modeling_bert.py:315
Methodsize_divisibility
(self)
grit/modeling/backbone/vit.py:424
Methodtextual_feature_size
(self)
grit/modeling/text/text_decoder.py:20
Functionwrapper
(url, *args, **kwargs)
grit/modeling/text/file_utils.py:142
← previous101–187 of 187, ranked by callers