Frank Lee 35813ed3c4 update examples and sphnix docs for the new api (#63)

2021-12-13 22:07:01 +08:00

..

data_sampler

update examples and sphnix docs for the new api (#63)

2021-12-13 22:07:01 +08:00

gradient_accumulation

update examples and sphnix docs for the new api (#63)

2021-12-13 22:07:01 +08:00

multi_tensor_apply

update examples and sphnix docs for the new api (#63)

2021-12-13 22:07:01 +08:00

__init__.py

Develop/experiments (#59)

2021-12-09 15:08:29 +08:00

activation_checkpoint.py

Migrated project

2021-10-28 18:21:23 +02:00

checkpointing.py

Support TP-compatible Torch AMP and Update trainer API (#27)

2021-11-18 19:45:06 +08:00

common.py

update examples and sphnix docs for the new api (#63)

2021-12-13 22:07:01 +08:00

cuda.py

Migrated project

2021-10-28 18:21:23 +02:00

memory.py

update examples and sphnix docs for the new api (#63)

2021-12-13 22:07:01 +08:00

timer.py

update examples and sphnix docs for the new api (#63)

2021-12-13 22:07:01 +08:00