18 lines
377 B
Python
18 lines
377 B
Python
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class TrainingStatus(BaseModel):
|
|
current_step: int
|
|
total_steps: int
|
|
loss: float
|
|
learning_rate: float
|
|
percentage: float
|
|
eta_seconds: Optional[float]
|
|
steps_per_second: float
|
|
updated_at: datetime
|
|
source: str # 'local' or 'remote'
|
|
source_path: str
|