Welcome to torch4keras’s documentation!

Home:

• Quick-Start
  • 1. 下载安装
  • 2. 功能
• History

API:

• Model
  • AccelerateTrainer
    • AccelerateTrainer.compile()
    • AccelerateTrainer.loss_backward()
    • AccelerateTrainer.prepare()
    • AccelerateTrainer.unwrap_model()
  • BaseModel
  • BaseModelDDP
  • BaseModelDP
  • DeepSpeedTrainer
    • DeepSpeedTrainer.compile()
    • DeepSpeedTrainer.loss_backward()
    • DeepSpeedTrainer.resume_from_checkpoint()
    • DeepSpeedTrainer.save_to_checkpoint()
    • DeepSpeedTrainer.step()
    • DeepSpeedTrainer.unwrap_model()
  • Trainer
    • Trainer.compile()
    • Trainer.device
    • Trainer.fit()
    • Trainer.load_steps_params()
    • Trainer.load_weights()
    • Trainer.loss_backward()
    • Trainer.predict()
    • Trainer.print_trainable_parameters()
    • Trainer.resume_from_checkpoint()
    • Trainer.save_steps_params()
    • Trainer.save_to_checkpoint()
    • Trainer.save_weights()
    • Trainer.step()
    • Trainer.to_model_device()
    • Trainer.unwrap_model()
  • TrainerDDP
  • TrainerDP
  • add_trainer()
• Snippets
  • DottableDict
  • IterDataset
  • ListDataset
  • colorful()
  • email_when_error()
  • get_parameter_device()
  • info_level_prefix()
  • metric_mapping()
  • print_trainable_parameters()
  • search_layer()
  • seed_everything()
  • send_email()
  • softmax()
  • spend_time()
  • take_along_dim()
  • torch_div()

Indices and tables

• Index

• Module Index

• Search Page