You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
14 lines
249 B
14 lines
249 B
4 years ago
|
from dataclasses import dataclass
|
||
|
|
|
||
|
|
from .logger import Logger
|
||
|
|
from timm.utils.checkpoint_saver import CheckpointSaver
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class TrainServices:
|
||
|
|
""" Train Loop Services
|
||
|
|
"""
|
||
|
|
logger: Logger = None
|
||
|
|
saver: CheckpointSaver = None
|
||
|
|
|