We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a37b8ef commit b8dc579Copy full SHA for b8dc579
internlm/core/trainer.py
@@ -6,13 +6,14 @@
6
import json
7
import math
8
import os
9
-from collections import deque
10
import time
+from collections import deque
11
from typing import Iterable, List, Optional
12
13
from torch.utils.data import DataLoader
14
15
-from internlm.core.context import ParallelMode, global_context as gpc
+from internlm.core.context import ParallelMode
16
+from internlm.core.context import global_context as gpc
17
from internlm.core.engine import Engine
18
from internlm.core.parallel.comm import ISPCommunicatorSchedulerHook
19
from internlm.core.scheduler import BaseScheduler, NonPipelineScheduler
0 commit comments