MCPcopy Create free account

hub / github.com/UIC-Liu-Lab/CPT / functions

Functions124 in github.com/UIC-Liu-Lab/CPT

↓ 18 callersMethod__init__
(self, config)
networks/adapter_mask/roberta_model.py:401
↓ 14 callersMethodtranspose_for_scores
(self, x)
networks/adapter_mask/roberta_model.py:194
↓ 6 callersMethodmask
(self, t: torch.LongTensor, s: int = None)
networks/adapter_mask/roberta_adapter.py:60
↓ 3 callersMethod__init__
(self, config: AdapterMaskConfig)
networks/adapter_mask/roberta_adapter.py:18
↓ 3 callersFunctionadapt_roberta_self_output
(config: AdapterMaskConfig)
networks/adapter_mask/roberta_adapter.py:213
↓ 3 callersMethodtrain
(self, model, accelerator, train_loader, test_loader)
approaches/finetune.py:28
↓ 2 callersFunctionget_dataset
(dataset_name, tokenizer, args)
dataloader/data.py:89
↓ 2 callersFunctionload_roberta_adapter_model
( roberta_model: nn.Module, checkpoint: str = None, mode: str = "sequential",
networks/adapter_mask/roberta_adapter.py:247
↓ 2 callersMethodupdate_keys_to_ignore
Remove some keys from ignore list
networks/adapter_mask/roberta_model.py:634
↓ 1 callersMethod__init__
(self, config)
networks/adapter_mask/roberta_base.py:47
↓ 1 callersMethod__init__
( self, device_placement: bool = True, split_batches: bool = False, fp16: bool = None, cpu: bool =
networks/adapter_mask/MyAccelerator.py:17
↓ 1 callersFunctionadapt_roberta_self_attn
(config: AdapterMaskConfig)
networks/adapter_mask/roberta_adapter.py:217
↓ 1 callersFunctionadd_roberta_adapters
(roberta_model: BertModel, config: AdapterMaskConfig)
networks/adapter_mask/roberta_adapter.py:221
↓ 1 callersMethodafter_training_op
(self, accelerator, model, tokenizer, mask_pre=None)
approaches/posttrain.py:217
↓ 1 callersFunctioncreate_position_ids_from_input_ids
Replace non-padding symbols with their position numbers. Position numbers begin at padding_idx+1. Padding symbols are ignored. This is modifi
networks/adapter_mask/roberta_model.py:1692
↓ 1 callersMethodcreate_position_ids_from_inputs_embeds
We are provided embeddings directly. We cannot infer which are padded so just generate sequential position ids. Args: in
networks/adapter_mask/roberta_model.py:150
↓ 1 callersMethodeval
(self, model, dataloader, accelerator)
approaches/finetune.py:195
↓ 1 callersFunctionfreeze_all_parameters
(model: nn.Module)
networks/adapter_mask/common.py:24
↓ 1 callersFunctionget_acl_unsup
()
dataloader/data.py:43
↓ 1 callersFunctionget_agnews_unsup
()
dataloader/data.py:36
↓ 1 callersFunctionget_ai_unsup
()
dataloader/data.py:50
↓ 1 callersMethodget_feature
(self, gfc1, gfc2, x)
networks/adapter_mask/roberta_adapter.py:51
↓ 1 callersFunctionget_restaurant_sup
(tokenizer)
dataloader/data.py:57
↓ 1 callersFunctionget_restaurant_unsup
()
dataloader/data.py:29
↓ 1 callersFunctionget_view_for
(model, n, p, masks)
networks/adapter_mask/roberta_adapter.py:359
↓ 1 callersFunctionlabel2idx
(label)
dataloader/data.py:60
↓ 1 callersFunctionmain
()
finetune.py:40
↓ 1 callersFunctionmain
()
posttrain.py:46
↓ 1 callersMethodmy_prepare
(self, *args)
networks/adapter_mask/MyAccelerator.py:38
↓ 1 callersMethodmy_prepare_one
(self, obj)
networks/adapter_mask/MyAccelerator.py:26
↓ 1 callersMethodmy_prepare_optimizer
(self, optimizer)
networks/adapter_mask/MyAccelerator.py:23
↓ 1 callersMethodmy_step
Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model
networks/adapter_mask/my_optimization.py:105
↓ 1 callersFunctionparsing_finetune
()
config.py:164
↓ 1 callersFunctionparsing_posttrain
()
config.py:12
↓ 1 callersMethodprune_heads
(self, heads)
networks/adapter_mask/roberta_model.py:321
↓ 1 callersMethodtrain_epoch
(self, model, optimizer, dataloader, accelerator, lr_scheduler)
approaches/finetune.py:166
↓ 1 callersFunctionunfreeze_roberta_adapters
(roberta_model: nn.Module)
networks/adapter_mask/roberta_adapter.py:238
Method__init__
(self, config: AdapterMaskConfig)
networks/adapter_mask/roberta_adapter.py:32
Method__init__
(self, self_output: BertSelfOutput, config: AdapterMaskConfig)
networks/adapter_mask/roberta_adapter.py:76
Method__init__
(self, self_attn: RobertaSelfAttention, config: AdapterMaskConfig)
networks/adapter_mask/roberta_adapter.py:101
Method__init__
(self, config)
networks/adapter_mask/roberta_base.py:52
Method__init__
(self, optimizer, device_placement=True, scaler=None)
networks/adapter_mask/MyAccelerator.py:8
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:84
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:170
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:297
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:315
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:367
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:383
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:487
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:589
Method__init__
(self, config, add_pooling_layer=True)
networks/adapter_mask/roberta_model.py:736
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:911
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:996
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:1151
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:1249
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:1286
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:1387
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:1482
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:1570
Method__init__
(self, config)
networks/adapter_mask/roberta_model.py:1600
Method__init__
(self, params, lr=required, warmup=-1, t_total=-1, schedule='warmup_linear', b1=0.9, b2=0.999, e=1e-6,
networks/adapter_mask/my_optimization.py:65
Method__init__
(self, args)
approaches/finetune.py:23
Method__init__
(self, args)
approaches/posttrain.py:29
Method__post_init__
(self)
networks/adapter_mask/common.py:19
Method_init_weights
Initialize the weights
networks/adapter_mask/roberta_model.py:614
Method_prune_heads
Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer} See base class PreTrainedModel
networks/adapter_mask/roberta_model.py:755
Method_reorder_cache
(self, past, beam_idx)
networks/adapter_mask/roberta_model.py:1138
Method_set_gradient_checkpointing
(self, module, value=False)
networks/adapter_mask/roberta_model.py:630
Method_tie_weights
(self)
networks/adapter_mask/roberta_model.py:1271
Functioncombine_function
(example)
dataloader/data.py:117
Methodcreate_custom_forward
(module)
networks/adapter_mask/roberta_model.py:527
Methodcustom_forward
(*inputs)
networks/adapter_mask/roberta_model.py:528
Methodeval
(self, model, eval_dataloader, eval_dataset, accelerator)
approaches/posttrain.py:257
Methodfeed_forward_chunk
(self, attention_output, **kwargs)
networks/adapter_mask/roberta_model.py:479
Functionforward
(self, t, input_ids, segment_ids, input_mask, s=None)
networks/adapter_mask/roberta_adapter.py:296
Methodforward
(self, x)
networks/adapter_mask/roberta_adapter.py:24
Methodforward
(self, x, t, s, smax=400, add_residual=True, residual=None)
networks/adapter_mask/roberta_adapter.py:40
Methodforward
(self, hidden_states: torch.Tensor, input_tensor: torch.Tensor, t, s, **kwargs)
networks/adapter_mask/roberta_adapter.py:84
Methodforward
( self, hidden_states, attention_mask=None, head_mask=None,
networks/adapter_mask/roberta_adapter.py:110
Methodforward
( self, input_ids=None, past_key_values=None, attention_mask=None, tok
networks/adapter_mask/roberta_base.py:15
Methodforward
( self, input_ids=None, token_type_ids=None, position_ids=None, inputs_embeds=None, past_key_value
networks/adapter_mask/roberta_model.py:110
Methodforward
( self, hidden_states, attention_mask=None, head_mask=None,
networks/adapter_mask/roberta_model.py:199
Methodforward
(self, hidden_states, input_tensor, **kwargs)
networks/adapter_mask/roberta_model.py:306
Methodforward
( self, hidden_states, attention_mask=None, head_mask=None,
networks/adapter_mask/roberta_model.py:339
Methodforward
(self, hidden_states)
networks/adapter_mask/roberta_model.py:375
Methodforward
(self, hidden_states, input_tensor, **kwargs)
networks/adapter_mask/roberta_model.py:392
Methodforward
( self, hidden_states, attention_mask=None, head_mask=None,
networks/adapter_mask/roberta_model.py:414
Methodforward
( self, hidden_states, attention_mask=None, head_mask=None,
networks/adapter_mask/roberta_model.py:493
Methodforward
(self, hidden_states)
networks/adapter_mask/roberta_model.py:594
Methodforward
r""" encoder_hidden_states (:obj:`torch.FloatTensor` of shape :obj:`(batch_size, sequence_length, hidden_size)`, `optional`): Seq
networks/adapter_mask/roberta_model.py:771
Methodforward
( self, input_ids=None, attention_mask=None, token_type_ids=No
networks/adapter_mask/roberta_model.py:927
Methodforward
r""" encoder_hidden_states (:obj:`torch.FloatTensor` of shape :obj:`(batch_size, sequence_length, hidden_size)`, `optional`): Seq
networks/adapter_mask/roberta_model.py:1018
Methodforward
r""" labels (:obj:`torch.LongTensor` of shape :obj:`(batch_size, sequence_length)`, `optional`): Labels for computing the masked l
networks/adapter_mask/roberta_model.py:1182
Methodforward
(self, features, inv_lang_adapter=None, **kwargs)
networks/adapter_mask/roberta_model.py:1258
Methodforward
r""" labels (:obj:`torch.LongTensor` of shape :obj:`(batch_size,)`, `optional`): Labels for computing the sequence classification/
networks/adapter_mask/roberta_model.py:1303
Methodforward
r""" labels (:obj:`torch.LongTensor` of shape :obj:`(batch_size,)`, `optional`): Labels for computing the multiple choice classifi
networks/adapter_mask/roberta_model.py:1403
Methodforward
r""" labels (:obj:`torch.LongTensor` of shape :obj:`(batch_size, sequence_length)`, `optional`): Labels for computing the token cl
networks/adapter_mask/roberta_model.py:1502
Methodforward
(self, features, **kwargs)
networks/adapter_mask/roberta_model.py:1579
Methodforward
r""" start_positions (:obj:`torch.LongTensor` of shape :obj:`(batch_size,)`, `optional`): Labels for position (index) of the start
networks/adapter_mask/roberta_model.py:1616
Functionforward_cls
(self, t, input_ids, segment_ids, input_mask, start_mixup=None, s=None, l=None, idx=None, mix_type=None)
networks/adapter_mask/roberta_adapter.py:311
next →1–100 of 124, ranked by callers