15 lines
295 B
Python
15 lines
295 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
|
|
eta_seconds: Optional[float]
|
|
steps_per_second: float
|
|
updated_at: datetime
|