mirror of
https://github.com/twitter/the-algorithm-ml.git
synced 2024-12-23 06:41:49 +01:00
Update custom_training_loop.py
formatted imports
This commit is contained in:
parent
415863550d
commit
7d9ac9a643
@ -8,14 +8,22 @@ Features (go/100_enablements):
|
|||||||
- go/dataset-service 0-copy integration
|
- go/dataset-service 0-copy integration
|
||||||
|
|
||||||
"""
|
"""
|
||||||
import datetime
|
|
||||||
import os
|
import os
|
||||||
from typing import Callable, Dict, Iterable, List, Mapping, Optional
|
import datetime
|
||||||
|
from typing import (
|
||||||
|
Callable,
|
||||||
|
Dict,
|
||||||
|
Iterable,
|
||||||
|
List,
|
||||||
|
Mapping,
|
||||||
|
Optional
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
from tml.common import log_weights
|
from tml.common import log_weights
|
||||||
import tml.common.checkpointing.snapshot as snapshot_lib
|
|
||||||
from tml.core.losses import get_global_loss_detached
|
from tml.core.losses import get_global_loss_detached
|
||||||
|
import tml.common.checkpointing.snapshot as snapshot_lib
|
||||||
from tml.ml_logging.torch_logging import logging # type: ignore[attr-defined]
|
from tml.ml_logging.torch_logging import logging # type: ignore[attr-defined]
|
||||||
from tml.core.train_pipeline import TrainPipelineSparseDist
|
from tml.core.train_pipeline import TrainPipelineSparseDist
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user