mirror of
https://github.com/OpenBMB/VoxCPM
synced 2025-12-14 20:38:12 +00:00
Update: VoxCPM1.5 and fine-tuning supprt
This commit is contained in:
21
src/voxcpm/training/state.py
Normal file
21
src/voxcpm/training/state.py
Normal file
@@ -0,0 +1,21 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from dataclasses import dataclass
|
||||
|
||||
|
||||
@dataclass
|
||||
class TrainingState:
|
||||
"""
|
||||
Container that mirrors the object returned in the minicpm-audio training
|
||||
loop. It holds persistent references to the model, optimizer, scheduler,
|
||||
dataloaders and tracker.
|
||||
"""
|
||||
|
||||
generator: object
|
||||
optimizer: object
|
||||
scheduler: object
|
||||
train_loader: object
|
||||
val_loader: object
|
||||
tracker: object
|
||||
batch_processor: object
|
||||
|
||||
Reference in New Issue
Block a user