MCPcopy Create free account

hub / github.com/Pandalin98/ITFormer-ICML25 / functions

Functions129 in github.com/Pandalin98/ITFormer-ICML25

⨍ Functions 129◇ Types & classes 38
Methodforward
(self, x, query=None)
models/TimeSeriesEncoder.py:51
Methodforward
(self, x)
models/TimeSeriesEncoder.py:118
Methodforward
(self, x, offset=0)
models/TimeSeriesEncoder.py:151
Methodforward
(self, x, attn_mask=None)
models/TimeSeriesEncoder.py:180
Methodforward
(self, x)
models/TimeSeriesEncoder.py:225
Methodforward
(self, x)
models/TimeSeriesEncoder.py:285
Methodforward
(self, x, attn_mask)
models/TimeSeriesEncoder.py:320
Methodforward
(self, x)
models/TimeSeriesEncoder.py:363
Methodforward
(self, x, prefix_seq_len=None)
models/TimeSeriesEncoder.py:403
Methodforward
(self, x, prefix_seq_len, attn_mask)
models/TimeSeriesEncoder.py:443
Methodforward
(self, x)
models/TimeSeriesEncoder.py:456
Methodforward
(self, query, key_value, attn_mask=None)
models/ITFormer.py:40
Methodforward
(self, x, key_value, attn_mask)
models/ITFormer.py:103
Methodforward
(self, query, key_value, attn_mask=None)
models/ITFormer.py:140
Methodforward
(self, x, key_value, attn_mask)
models/ITFormer.py:203
Methodforward
(self, x, attn_mask=None)
models/ITFormer.py:237
Methodforward
(self, x, attn_mask=None)
models/ITFormer.py:282
Methodforward
(self, x,memory, attn_mask=None)
models/ITFormer.py:319
Methodforward
(self, x, memory, attn_mask=None)
models/ITFormer.py:375
Methodforward
(self, x, memory, stage=None,attn_mask=None)
models/ITFormer.py:424
Methodforward
Args: query (torch.Tensor): Query tensor of shape (batch_size, L_q, d). memory (torch.Tensor): Memory tensor of shape
models/layers/attention.py:35
Functiongenerate
(mode, image_input, text_input, max_new_tokens = 100, temperature = 0.0, top_k = None)
utils/gradio_vlm.py:22
Functionget_local_world_size
()
utils/dist_util.py:23
Methodget_stage_loss_statistics
分析不同stage的损失分布，用于调整权重 Args: dataloader: 数据加载器 num_samples: 分析的样本数量 Returns:
EXP/exp_instruct.py:403
Functionget_world_size
()
utils/dist_util.py:34
Functionis_master_process
()
utils/dist_util.py:7
Methodload_model
(self, checkpoint_path)
EXP/exp_instruct.py:110
Functionlog_in_local_rank_0
(*msg, type='info', used_logger=None)
utils/log_util.py:41
Methodprepare_inputs_for_generation
Prepare inputs for text generation. Args: input_ids: Input token IDs query_ids: Query token IDs p
models/TimeLanguageModel.py:276
← previous101–129 of 129, ranked by callers