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.
pytorch-image-models/timm/bits/train_services.py

14 lines
261 B

from dataclasses import dataclass
from .monitor import Monitor
from .checkpoint_manager import CheckpointManager
@dataclass
class TrainServices:
""" Train Loop Services
"""
logger: Monitor = None
checkpoint_manager: CheckpointManager = None