Frank Lee af88570f4b fixed bugs in compatibility between torch amp and tensor parallel and performed some minor fixes

2021-11-10 10:47:58 +08:00

..

builder

Add gradient accumulation, fix lr scheduler

2021-11-08 16:30:24 +08:00

communication

Migrated project

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

context

Migrated project

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

engine

fixed bugs in compatibility between torch amp and tensor parallel and performed some minor fixes

2021-11-10 10:47:58 +08:00

logging

Migrated project

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

nn

fixed bugs in compatibility between torch amp and tensor parallel and performed some minor fixes

2021-11-10 10:47:58 +08:00

registry

Migrated project

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

trainer

fixed bugs in compatibility between torch amp and tensor parallel and performed some minor fixes

2021-11-10 10:47:58 +08:00

utils

fixed bugs in compatibility between torch amp and tensor parallel and performed some minor fixes

2021-11-10 10:47:58 +08:00

__init__.py

Migrated project

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

checkpointing.py

Migrated project

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

constants.py

Migrated project

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

core.py

Migrated project

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

initialize.py

Add gradient accumulation, fix lr scheduler

2021-11-08 16:30:24 +08:00