Index

A

AccelerateTrainer (class in torch4keras.model)

add_trainer() (in module torch4keras.model)

B

BaseModel (class in torch4keras.model)

BaseModelDDP (class in torch4keras.model) BaseModelDP (class in torch4keras.model)

C

colorful() (in module torch4keras.snippets) compile() (torch4keras.model.AccelerateTrainer method) (torch4keras.model.DeepSpeedTrainer method) (torch4keras.model.Trainer method)

D

DeepSpeedTrainer (class in torch4keras.model)

device (torch4keras.model.Trainer property) DottableDict (class in torch4keras.snippets)

E

email_when_error() (in module torch4keras.snippets)

F

fit() (torch4keras.model.Trainer method)

G

get_parameter_device() (in module torch4keras.snippets)

I

info_level_prefix() (in module torch4keras.snippets)

IterDataset (class in torch4keras.snippets)

L

ListDataset (class in torch4keras.snippets) load_steps_params() (torch4keras.model.Trainer method) load_weights() (torch4keras.model.Trainer method)

loss_backward() (torch4keras.model.AccelerateTrainer method) (torch4keras.model.DeepSpeedTrainer method) (torch4keras.model.Trainer method)

M

metric_mapping() (in module torch4keras.snippets) module torch4keras.model torch4keras.snippets

P

predict() (torch4keras.model.Trainer method) prepare() (torch4keras.model.AccelerateTrainer method)

print_trainable_parameters() (in module torch4keras.snippets) (torch4keras.model.Trainer method)

R

resume_from_checkpoint() (torch4keras.model.DeepSpeedTrainer method) (torch4keras.model.Trainer method)

S

save_steps_params() (torch4keras.model.Trainer method) save_to_checkpoint() (torch4keras.model.DeepSpeedTrainer method) (torch4keras.model.Trainer method) save_weights() (torch4keras.model.Trainer method) search_layer() (in module torch4keras.snippets)

seed_everything() (in module torch4keras.snippets) send_email() (in module torch4keras.snippets) softmax() (in module torch4keras.snippets) spend_time() (in module torch4keras.snippets) step() (torch4keras.model.DeepSpeedTrainer method) (torch4keras.model.Trainer method)

T

take_along_dim() (in module torch4keras.snippets) to_model_device() (torch4keras.model.Trainer method) torch4keras.model module torch4keras.snippets module

torch_div() (in module torch4keras.snippets) Trainer (class in torch4keras.model) TrainerDDP (in module torch4keras.model) TrainerDP (in module torch4keras.model)

U

unwrap_model() (torch4keras.model.AccelerateTrainer method) (torch4keras.model.DeepSpeedTrainer method) (torch4keras.model.Trainer method)