Update model/trainer.py
Browse files- model/trainer.py +3 -3
model/trainer.py
CHANGED
@@ -14,9 +14,9 @@ from torch.optim.lr_scheduler import LinearLR, SequentialLR
|
|
14 |
from torch.utils.data import DataLoader, Dataset, SequentialSampler
|
15 |
from tqdm import tqdm
|
16 |
|
17 |
-
from
|
18 |
-
from
|
19 |
-
from
|
20 |
|
21 |
# trainer
|
22 |
|
|
|
14 |
from torch.utils.data import DataLoader, Dataset, SequentialSampler
|
15 |
from tqdm import tqdm
|
16 |
|
17 |
+
from model import CFM
|
18 |
+
from model.dataset import DynamicBatchSampler, collate_fn
|
19 |
+
from model.utils import default, exists
|
20 |
|
21 |
# trainer
|
22 |
|