Mirrors/the-algorithm-ml
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm-ml.git synced 2024-06-05 01:08:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update custom_training_loop.py

Browse Source 
formatted imports
This commit is contained in:
Siddhi Kiran Bajracharyya 2023-04-03 18:00:58 -05:00 committed by GitHub
parent 415863550d
commit 7d9ac9a643
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
core/custom_training_loop.py
View File

@ -8,14 +8,22 @@ Features (go/100_enablements):
- go/dataset-service 0-copy integration
"""
import datetime
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
import tml.common.checkpointing.snapshot as snapshot_lib
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.core.train_pipeline import TrainPipelineSparseDist