pytorch-accelerated
latest
Get Started:
Installation
Quickstart
API Reference:
Trainer
Callbacks
Tracking
Run Config
Fine-tuning
Schedulers
Utils
pytorch-accelerated
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
E
|
F
|
G
|
I
|
L
|
M
|
O
|
P
|
R
|
S
|
T
|
U
_
__init__() (pytorch_accelerated.callbacks.EarlyStoppingCallback method)
(pytorch_accelerated.callbacks.ModelEmaCallback method)
(pytorch_accelerated.callbacks.SaveBestModelCallback method)
(pytorch_accelerated.schedulers.cosine_scheduler.CosineLrScheduler method)
(pytorch_accelerated.schedulers.scheduler_base.SchedulerBase method)
(pytorch_accelerated.schedulers.scheduler_base.StatefulSchedulerBase method)
(pytorch_accelerated.trainer.Trainer method)
_clip_gradients() (pytorch_accelerated.trainer.Trainer method)
_create_accelerator() (pytorch_accelerated.trainer.Trainer method)
_create_run_config() (pytorch_accelerated.trainer.Trainer method)
_prepare_model_optimizer_and_dataloaders() (pytorch_accelerated.trainer.Trainer method)
_run_eval_epoch() (pytorch_accelerated.trainer.Trainer method)
_run_train_epoch() (pytorch_accelerated.trainer.Trainer method)
_run_training() (pytorch_accelerated.trainer.Trainer method)
A
add_callback() (pytorch_accelerated.callbacks.CallbackHandler method)
add_callbacks() (pytorch_accelerated.callbacks.CallbackHandler method)
B
backward_step() (pytorch_accelerated.trainer.Trainer method)
C
calculate_eval_batch_loss() (pytorch_accelerated.trainer.Trainer method)
calculate_train_batch_loss() (pytorch_accelerated.trainer.Trainer method)
call_event() (pytorch_accelerated.callbacks.CallbackHandler method)
CallbackHandler (class in pytorch_accelerated.callbacks)
ConvertSyncBatchNormCallback (class in pytorch_accelerated.callbacks)
CosineLrScheduler (class in pytorch_accelerated.schedulers.cosine_scheduler)
create_eval_dataloader() (pytorch_accelerated.trainer.Trainer method)
create_scheduler() (pytorch_accelerated.trainer.Trainer method)
create_scheduler_fn() (pytorch_accelerated.schedulers.cosine_scheduler.CosineLrScheduler class method)
create_train_dataloader() (pytorch_accelerated.trainer.Trainer method)
current_epoch (pytorch_accelerated.tracking.RunHistory property)
E
EarlyStoppingCallback (class in pytorch_accelerated.callbacks)
eval_epoch_end() (pytorch_accelerated.trainer.Trainer method)
eval_epoch_start() (pytorch_accelerated.trainer.Trainer method)
evaluate() (pytorch_accelerated.trainer.Trainer method)
evaluation_run_end() (pytorch_accelerated.trainer.Trainer method)
evaluation_run_start() (pytorch_accelerated.trainer.Trainer method)
F
freeze() (pytorch_accelerated.finetuning.ModelFreezer method)
G
gather() (pytorch_accelerated.trainer.Trainer method)
get_default_eval_dl_kwargs() (pytorch_accelerated.trainer.Trainer method)
get_default_train_dl_kwargs() (pytorch_accelerated.trainer.Trainer method)
get_latest_metric() (pytorch_accelerated.tracking.RunHistory method)
get_layer_groups() (pytorch_accelerated.finetuning.ModelFreezer method)
get_layers() (pytorch_accelerated.finetuning.ModelFreezer method)
get_metric_names() (pytorch_accelerated.tracking.RunHistory method)
get_metric_values() (pytorch_accelerated.tracking.RunHistory method)
get_model() (pytorch_accelerated.trainer.Trainer method)
get_trainable_parameters() (pytorch_accelerated.finetuning.ModelFreezer method)
get_updated_values() (pytorch_accelerated.schedulers.cosine_scheduler.CosineLrScheduler method)
(pytorch_accelerated.schedulers.scheduler_base.SchedulerBase method)
I
InMemoryRunHistory (class in pytorch_accelerated.tracking)
L
load_checkpoint() (pytorch_accelerated.trainer.Trainer method)
load_state_dict() (pytorch_accelerated.schedulers.scheduler_base.SchedulerBase method)
LogMetricsCallback (class in pytorch_accelerated.callbacks)
M
metric_name_prefix (pytorch_accelerated.tracking.RunHistory property)
ModelEma (class in pytorch_accelerated.utils)
ModelEmaCallback (class in pytorch_accelerated.callbacks)
ModelFreezer (class in pytorch_accelerated.finetuning)
MoveModulesToDeviceCallback (class in pytorch_accelerated.callbacks)
O
on_eval_epoch_end() (pytorch_accelerated.callbacks.TrainerCallback method)
on_eval_epoch_start() (pytorch_accelerated.callbacks.TrainerCallback method)
on_eval_step_end() (pytorch_accelerated.callbacks.TrainerCallback method)
on_eval_step_start() (pytorch_accelerated.callbacks.TrainerCallback method)
on_init_end() (pytorch_accelerated.callbacks.TrainerCallback method)
on_stop_training_error() (pytorch_accelerated.callbacks.TrainerCallback method)
on_train_epoch_end() (pytorch_accelerated.callbacks.TrainerCallback method)
on_train_epoch_start() (pytorch_accelerated.callbacks.TrainerCallback method)
on_train_step_end() (pytorch_accelerated.callbacks.TrainerCallback method)
on_train_step_start() (pytorch_accelerated.callbacks.TrainerCallback method)
on_training_run_end() (pytorch_accelerated.callbacks.TrainerCallback method)
on_training_run_start() (pytorch_accelerated.callbacks.TrainerCallback method)
optimizer_step() (pytorch_accelerated.trainer.Trainer method)
optimizer_zero_grad() (pytorch_accelerated.trainer.Trainer method)
P
print() (pytorch_accelerated.trainer.Trainer method)
PrintProgressCallback (class in pytorch_accelerated.callbacks)
ProgressBarCallback (class in pytorch_accelerated.callbacks)
R
reset() (pytorch_accelerated.tracking.RunHistory method)
RunHistory (class in pytorch_accelerated.tracking)
S
save_checkpoint() (pytorch_accelerated.trainer.Trainer method)
SaveBestModelCallback (class in pytorch_accelerated.callbacks)
scheduler_step() (pytorch_accelerated.trainer.Trainer method)
SchedulerBase (class in pytorch_accelerated.schedulers.scheduler_base)
set_metric_name_prefix() (pytorch_accelerated.tracking.RunHistory method)
state_dict() (pytorch_accelerated.schedulers.scheduler_base.SchedulerBase method)
StatefulSchedulerBase (class in pytorch_accelerated.schedulers.scheduler_base)
step() (pytorch_accelerated.schedulers.scheduler_base.StatefulSchedulerBase method)
step_update() (pytorch_accelerated.schedulers.scheduler_base.SchedulerBase method)
T
TerminateOnNaNCallback (class in pytorch_accelerated.callbacks)
train() (pytorch_accelerated.trainer.Trainer method)
train_epoch_end() (pytorch_accelerated.trainer.Trainer method)
train_epoch_start() (pytorch_accelerated.trainer.Trainer method)
Trainer (class in pytorch_accelerated.trainer)
TrainerCallback (class in pytorch_accelerated.callbacks)
TrainerPlaceholderValues (class in pytorch_accelerated.trainer)
TrainerRunConfig (class in pytorch_accelerated.run_config)
TrainerWithTimmScheduler (class in pytorch_accelerated.trainer)
training_run_end() (pytorch_accelerated.trainer.Trainer method)
training_run_epoch_end() (pytorch_accelerated.trainer.Trainer method)
training_run_start() (pytorch_accelerated.trainer.Trainer method)
U
unfreeze() (pytorch_accelerated.finetuning.ModelFreezer method)
update_metric() (pytorch_accelerated.tracking.RunHistory method)
Read the Docs
v: latest
Versions
latest
Downloads
pdf
On Read the Docs
Project Home
Builds