Method__init__(
self, data_path, data_split, data_repeat, node_rank, world_size, log_dir
)
sgs/verification/prover/workers/data_loader.py:10
Method__init__(
self,
local_rank,
node_rank,
model_path,
task_queue,
request
sgs/verification/prover/workers/generator.py:11
Method__init__(self, idx, log_dir, tokenizer_path, scheduler, data_loader, cfg)
sgs/verification/prover/workers/search.py:13
Method__init__(self, idx, task_queue, request_statuses, lock, extra_args=AttrDict())
sgs/verification/prover/lean/verifier.py:271
Method__init__(
self, max_concurrent_requests=64, timeout=300, memory_limit=-1, name="verifier", lean_version="4.15"
sgs/verification/prover/lean/verifier.py:656
Method__init__(self, scheduler, tokenizer_path, process_print, cfg, **kwargs)
sgs/verification/prover/algorithms/base.py:10
Methodcompute_loss(
self,
model: nn.Module,
inputs: Dict[str, Union[torch.Tensor, Any]],
return_
sgs/training/custom_trainers.py:57
Methodcompute_loss( # Fixed typo
self,
model: nn.Module,
inputs: Dict[str, Union[torch.Tensor, Any]],
sgs/training/custom_trainers.py:135
Methodcompute_loss( # Fixed typo
self,
model: nn.Module,
inputs: Dict[str, Union[torch.Tensor, Any]],
sgs/training/custom_trainers.py:431