MCPcopy Create free account

hub / github.com/Time-MoE/Time-MoE / functions

Functions134 in github.com/Time-MoE/Time-MoE

⨍ Functions 134◇ Types & classes 30
Method_reorder_cache
(past_key_values, beam_idx)
time_moe/models/modeling_time_moe.py:1166
Method_sample
( self, input_ids: torch.LongTensor, logits_processor: LogitsProcessorList
time_moe/models/ts_generation_mixin.py:14
Method_set_signature_columns_if_needed
(self)
time_moe/trainer/hf_trainer.py:45
Method_upad_input
(self, query_layer, key_layer, value_layer, attention_mask, query_length)
time_moe/models/modeling_time_moe.py:614
Methodcreate_scheduler
(self, num_training_steps: int, optimizer: torch.optim.Optimizer = None)
time_moe/trainer/hf_trainer.py:24
Methodforward
(self, x)
time_moe/models/modeling_time_moe.py:186
Methodforward
(self, x, seq_len=None)
time_moe/models/modeling_time_moe.py:217
Methodforward
(self, hidden_states)
time_moe/models/modeling_time_moe.py:235
Methodforward
(self, hidden_state)
time_moe/models/modeling_time_moe.py:253
Methodforward
(self, hidden_state)
time_moe/models/modeling_time_moe.py:261
Methodforward
time_moe/models/modeling_time_moe.py:293
Methodforward
( self, hidden_states: torch.Tensor, attention_mask: Optional[torch.Tensor
time_moe/models/modeling_time_moe.py:383
Methodforward
( self, hidden_states: torch.Tensor, attention_mask: Optional[torch.LongTe
time_moe/models/modeling_time_moe.py:470
Methodforward
( self, hidden_states: torch.Tensor, attention_mask: Optional[torch.Tensor
time_moe/models/modeling_time_moe.py:678
Methodforward
( self, input_ids: torch.FloatTensor = None, attention_mask: Optional[torc
time_moe/models/modeling_time_moe.py:779
Methodforward
Args: x (torch.FloatTensor): with shape [B, seq_len, hidden_size] Returns: ` torch.FloatTensor: final predict
time_moe/models/modeling_time_moe.py:927
Methodforward
( self, input_ids: torch.FloatTensor = None, attention_mask: Optional[torc
time_moe/models/modeling_time_moe.py:974
Methodget_decoder
(self)
time_moe/models/modeling_time_moe.py:971
Functionget_local_world_size
()
time_moe/utils/dist_util.py:23
Methodget_num_tokens
(self)
time_moe/datasets/time_moe_dataset.py:89
Methodget_num_tokens
(self)
time_moe/datasets/binary_dataset.py:62
Methodget_num_tokens
(self)
time_moe/datasets/ts_dataset.py:16
Methodget_num_tokens
(self)
time_moe/datasets/general_dataset.py:27
Methodget_sequence_length_by_idx
(self, seq_idx)
time_moe/datasets/time_moe_dataset.py:79
Methodget_sequence_length_by_idx
(self, seq_idx)
time_moe/datasets/ts_dataset.py:20
Methodget_sequence_length_by_idx
(self, seq_idx)
time_moe/datasets/general_dataset.py:32
Functionis_master_process
()
time_moe/utils/dist_util.py:7
Methodis_valid_path
(data_path)
time_moe/datasets/ts_dataset.py:24
Methodis_valid_path
(data_path)
time_moe/datasets/general_dataset.py:37
Functionmax_scaler
(seq)
time_moe/datasets/time_moe_dataset.py:112
Functionparse_arbitrary_args
Parse arbitrary command-line arguments in the format --key value. Args: argv: List of arguments (typically sys.argv). Returns:
torch_dist_run.py:14
Functionsave_json_file
(obj, fn)
time_moe/datasets/binary_dataset.py:110
Methodset_decoder
(self, decoder)
time_moe/models/modeling_time_moe.py:968
Functionzero_scaler
(seq)
time_moe/datasets/time_moe_dataset.py:96
← previous101–134 of 134, ranked by callers