mirror of https://github.com/InternLM/InternLM
fix(train.py): fix ci lint error
parent
f3f2511e74
commit
c5a7e76ada
1
train.py
1
train.py
|
@ -27,7 +27,6 @@ from internlm.train import (
|
||||||
initialize_optimizer,
|
initialize_optimizer,
|
||||||
load_new_batch,
|
load_new_batch,
|
||||||
record_current_batch_training_metrics,
|
record_current_batch_training_metrics,
|
||||||
wrap_FSDP_model,
|
|
||||||
)
|
)
|
||||||
from internlm.utils.common import (
|
from internlm.utils.common import (
|
||||||
BatchSkipper,
|
BatchSkipper,
|
||||||
|
|
Loading…
Reference in New Issue