You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Xuanlei Zhao f71e63b0f3 [moe] support optimizer checkpoint (#5015) ... * Refactor MoE Manager setup method * unshard optim ckpt * optim io * update transformer version * update requirements * update ckpt * update ckpt * update ckpt * fix engine * fix engine		1 year ago
..
__init__.py	[moe] support optimizer checkpoint (#5015)	1 year ago
_operation.py	…
checkpoint.py	[moe] support optimizer checkpoint (#5015)	1 year ago
experts.py	[moe] support optimizer checkpoint (#5015)	1 year ago
layers.py	…
load_balance.py	…
loss.py	…
manager.py	[moe] support optimizer checkpoint (#5015)	1 year ago
routers.py	…
utils.py	…