SYMBOL INDEX (106 symbols across 13 files)

FILE: api.py
  function torch_gc (line 11) | def torch_gc():
  function create_item (line 22) | async def create_item(request: Request):

FILE: cli_demo.py
  function build_prompt (line 16) | def build_prompt(history):
  function signal_handler (line 24) | def signal_handler(signal, frame):
  function main (line 29) | def main():

FILE: cli_demo_vision.py
  function build_prompt (line 18) | def build_prompt(history, prefix):
  function signal_handler (line 26) | def signal_handler(signal, frame):
  function main (line 31) | def main():

FILE: ptuning/arguments.py
  class ModelArguments (line 6) | class ModelArguments:
  class DataTrainingArguments (line 65) | class DataTrainingArguments:
    method __post_init__ (line 212) | def __post_init__(self):

FILE: ptuning/main.py
  function main (line 49) | def main():
  function _mp_fn (line 424) | def _mp_fn(index):

FILE: ptuning/trainer.py
  class Trainer (line 222) | class Trainer:
    method __init__ (line 307) | def __init__(
    method add_callback (line 679) | def add_callback(self, callback):
    method pop_callback (line 690) | def pop_callback(self, callback):
    method remove_callback (line 706) | def remove_callback(self, callback):
    method _move_model_to_device (line 717) | def _move_model_to_device(self, model, device):
    method _set_signature_columns_if_needed (line 723) | def _set_signature_columns_if_needed(self):
    method _remove_unused_columns (line 731) | def _remove_unused_columns(self, dataset: "datasets.Dataset", descript...
    method _get_collator_with_removed_columns (line 757) | def _get_collator_with_removed_columns(
    method _get_train_sampler (line 775) | def _get_train_sampler(self) -> Optional[torch.utils.data.Sampler]:
    method get_train_dataloader (line 846) | def get_train_dataloader(self) -> DataLoader:
    method _get_eval_sampler (line 896) | def _get_eval_sampler(self, eval_dataset: Dataset) -> Optional[torch.u...
    method get_eval_dataloader (line 925) | def get_eval_dataloader(self, eval_dataset: Optional[Dataset] = None) ...
    method get_test_dataloader (line 975) | def get_test_dataloader(self, test_dataset: Dataset) -> DataLoader:
    method create_optimizer_and_scheduler (line 1023) | def create_optimizer_and_scheduler(self, num_training_steps: int):
    method create_optimizer (line 1039) | def create_optimizer(self):
    method get_optimizer_cls_and_kwargs (line 1096) | def get_optimizer_cls_and_kwargs(args: TrainingArguments) -> Tuple[Any...
    method create_scheduler (line 1186) | def create_scheduler(self, num_training_steps: int, optimizer: torch.o...
    method num_examples (line 1203) | def num_examples(self, dataloader: DataLoader) -> int:
    method _hp_search_setup (line 1217) | def _hp_search_setup(self, trial: Union["optuna.Trial", Dict[str, Any]]):
    method _report_to_hp_search (line 1258) | def _report_to_hp_search(self, trial: Union["optuna.Trial", Dict[str, ...
    method _tune_save_checkpoint (line 1276) | def _tune_save_checkpoint(self):
    method call_model_init (line 1289) | def call_model_init(self, trial=None):
    method torch_jit_model_eval (line 1303) | def torch_jit_model_eval(self, model, dataloader, training=False):
    method ipex_optimize_model (line 1341) | def ipex_optimize_model(self, model, training=False, dtype=torch.float...
    method _wrap_model (line 1364) | def _wrap_model(self, model, training=True, dataloader=None):
    method train (line 1556) | def train(
    method _inner_training_loop (line 1642) | def _inner_training_loop(
    method _get_output_dir (line 2055) | def _get_output_dir(self, trial):
    method _load_from_checkpoint (line 2075) | def _load_from_checkpoint(self, resume_from_checkpoint, model=None):
    method _load_best_model (line 2133) | def _load_best_model(self):
    method _issue_warnings_after_load (line 2193) | def _issue_warnings_after_load(self, load_result):
    method _maybe_log_save_evaluate (line 2206) | def _maybe_log_save_evaluate(self, tr_loss, model, trial, epoch, ignor...
    method _load_rng_state (line 2245) | def _load_rng_state(self, checkpoint):
    method _save_checkpoint (line 2286) | def _save_checkpoint(self, model, trial, metrics=None):
    method _load_optimizer_and_scheduler (line 2392) | def _load_optimizer_and_scheduler(self, checkpoint):
    method hyperparameter_search (line 2449) | def hyperparameter_search(
    method log (line 2545) | def log(self, logs: Dict[str, float]) -> None:
    method _prepare_input (line 2562) | def _prepare_input(self, data: Union[torch.Tensor, Any]) -> Union[torc...
    method _prepare_inputs (line 2580) | def _prepare_inputs(self, inputs: Dict[str, Union[torch.Tensor, Any]])...
    method compute_loss_context_manager (line 2596) | def compute_loss_context_manager(self):
    method autocast_smart_context_manager (line 2602) | def autocast_smart_context_manager(self, cache_enabled: Optional[bool]...
    method training_step (line 2621) | def training_step(self, model: nn.Module, inputs: Dict[str, Union[torc...
    method compute_loss (line 2669) | def compute_loss(self, model, inputs, return_outputs=False):
    method is_local_process_zero (line 2701) | def is_local_process_zero(self) -> bool:
    method is_world_process_zero (line 2708) | def is_world_process_zero(self) -> bool:
    method save_model (line 2720) | def save_model(self, output_dir: Optional[str] = None, _internal_call:...
    method _save_tpu (line 2784) | def _save_tpu(self, output_dir: Optional[str] = None):
    method _save (line 2812) | def _save(self, output_dir: Optional[str] = None, state_dict=None):
    method store_flos (line 2847) | def store_flos(self):
    method _sorted_checkpoints (line 2858) | def _sorted_checkpoints(
    method _rotate_checkpoints (line 2882) | def _rotate_checkpoints(self, use_mtime=False, output_dir=None) -> None:
    method evaluate (line 2907) | def evaluate(
    method predict (line 2978) | def predict(
    method evaluation_loop (line 3040) | def evaluation_loop(
    method _nested_gather (line 3251) | def _nested_gather(self, tensors, name=None):
    method _pad_across_processes (line 3269) | def _pad_across_processes(self, tensor, pad_index=-100):
    method prediction_step (line 3303) | def prediction_step(
    method floating_point_ops (line 3408) | def floating_point_ops(self, inputs: Dict[str, Union[torch.Tensor, Any...
    method init_git_repo (line 3426) | def init_git_repo(self, at_init: bool = False):
    method create_model_card (line 3473) | def create_model_card(
    method _push_from_checkpoint (line 3529) | def _push_from_checkpoint(self, checkpoint_folder):
    method push_to_hub (line 3571) | def push_to_hub(self, commit_message: Optional[str] = "End of training...
    method prediction_loop (line 3631) | def prediction_loop(
    method _gather_and_numpify (line 3777) | def _gather_and_numpify(self, tensors, name):
    method _add_sm_patterns_to_gitignore (line 3793) | def _add_sm_patterns_to_gitignore(self) -> None:

FILE: ptuning/trainer_seq2seq.py
  class Seq2SeqTrainer (line 30) | class Seq2SeqTrainer(Trainer):
    method evaluate (line 31) | def evaluate(
    method predict (line 80) | def predict(
    method prediction_step (line 138) | def prediction_step(
    method _pad_tensors_to_max_len (line 231) | def _pad_tensors_to_max_len(self, tensor, max_length):

FILE: ptuning/web_demo.py
  function postprocess (line 28) | def postprocess(self, y):
  function parse_text (line 42) | def parse_text(text):
  function predict (line 75) | def predict(input, chatbot, max_length, top_p, temperature, history):
  function reset_user_input (line 84) | def reset_user_input():
  function reset_state (line 88) | def reset_state():
  function main (line 119) | def main():

FILE: utils.py
  function auto_configure_device_map (line 8) | def auto_configure_device_map(num_gpus: int) -> Dict[str, int]:
  function load_model_on_gpus (line 38) | def load_model_on_gpus(checkpoint_path: Union[str, os.PathLike], num_gpu...

FILE: web_demo.py
  function postprocess (line 12) | def postprocess(self, y):
  function parse_text (line 26) | def parse_text(text):
  function predict (line 59) | def predict(input, chatbot, max_length, top_p, temperature, history):
  function reset_user_input (line 68) | def reset_user_input():
  function reset_state (line 72) | def reset_state():

FILE: web_demo2.py
  function get_model (line 13) | def get_model():
  function predict (line 24) | def predict(input, max_length, top_p, temperature, history=None):

FILE: web_demo_old.py
  function predict (line 12) | def predict(input, max_length, top_p, temperature, history=None):

FILE: web_demo_vision.py
  function postprocess (line 12) | def postprocess(self, y):
  function parse_text (line 26) | def parse_text(text):
  function predict (line 59) | def predict(input, image_path, chatbot, max_length, top_p, temperature, ...
  function predict_new_image (line 70) | def predict_new_image(image_path, chatbot, max_length, top_p, temperature):
  function reset_user_input (line 81) | def reset_user_input():
  function reset_state (line 85) | def reset_state():