Functions — github.com/Ugness/ELF-pytorch

CodeHub MCPcopy Create free account

hub / github.com/Ugness/ELF-pytorch / functions

Functions106 in github.com/Ugness/ELF-pytorch

⨍ Functions 106◇ Types & classes 19

Methodparity_check

Run the adapter on `(input_ids, attention_mask)` and compare against `reference_outputs`. Returns a dict with `max_abs_diff`, `mean_abs_diff`,

pytorch_lightning/encoders/base.py:57

(self, stage=None)

pytorch_lightning/lightning_module.py:107

(self, stage=None)

pytorch_lightning/lightning_module.py:360

Functionshift_left

(x: torch.Tensor, shift_per_sample: torch.Tensor, pad_value: int = 0, axis: int = 1)

pytorch_lightning/utils/generation_utils.py:20

Methodtrain_dataloader

pytorch_lightning/lightning_module.py:364

Methodtraining_step

(self, batch: Dict[str, torch.Tensor], batch_idx: int)

pytorch_lightning/lightning_module.py:140

← previous101–106 of 106, ranked by callers