ver217 dbe62c67b8 add an example of ViT-B/16 and remove w_norm clipping in LAMB (#29)

2021-11-18 23:45:09 +08:00

..

builder

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

2021-11-18 19:45:06 +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

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

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

logging

Migrated project

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

nn

add an example of ViT-B/16 and remove w_norm clipping in LAMB (#29)

2021-11-18 23:45:09 +08:00

registry

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

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

trainer

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

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

utils

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

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

__init__.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

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

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