Frank Lee ae035d305d [shardformer] added embedding gradient check (#4124)		2023-07-04 16:05:01 +08:00
..
__init__.py	[shardformer] init shardformer code structure (#3731)	2023-07-04 16:05:01 +08:00
autopolicy.py	[shardformer] import huggingface implicitly (#4101)	2023-07-04 16:05:01 +08:00
basepolicy.py	[shardformer] import huggingface implicitly (#4101)	2023-07-04 16:05:01 +08:00
bert.py	[shardformer] added embedding gradient check (#4124)	2023-07-04 16:05:01 +08:00
bloom.py	[shardformer] added embedding gradient check (#4124)	2023-07-04 16:05:01 +08:00
gpt2.py	[shardformer] import huggingface implicitly (#4101)	2023-07-04 16:05:01 +08:00
llama.py	[shardformer] import huggingface implicitly (#4101)	2023-07-04 16:05:01 +08:00
opt.py	[shardformer] added embedding gradient check (#4124)	2023-07-04 16:05:01 +08:00
t5.py	[shardformer] added embedding gradient check (#4124)	2023-07-04 16:05:01 +08:00
vit.py	[shardformer] import huggingface implicitly (#4101)	2023-07-04 16:05:01 +08:00