Add basic backend infrastructure

Signed-off-by: Felipe Cardoso <felipe.cardoso@hotmail.it>
This commit is contained in:
2025-01-22 18:02:07 +01:00
parent 80c8537614
commit 805ed647cd
16 changed files with 179 additions and 0 deletions

View File

View File

@@ -0,0 +1,11 @@
from datetime import datetime
from typing import Optional
from pydantic import BaseModel
class Sample(BaseModel):
filename: str
url: str
created_at: datetime
step: Optional[int] = None

View File

@@ -0,0 +1,14 @@
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