gitadmin/foxhunt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

proto(ml_training): add TrainingMode enum for fine-tune support

Browse Source 
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
jgrusewski
2026-03-01 21:32:01 +01:00
parent 6157ea0f77
commit ae92d2b99b
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
services/ml_training_service/proto/ml_training.proto
View File

@@ -67,6 +67,12 @@ service MLTrainingService {
// --- Core Request/Response Messages ---
// Training mode selection
enum TrainingMode {
TRAINING_MODE_FULL = 0; // Full training from scratch (default, backward-compatible)
TRAINING_MODE_FINE_TUNE = 1; // Fine-tune from existing checkpoint
}
// Request to start a new model training job
message StartTrainingRequest {
string model_type = 1; // Model type ("TLOB", "MAMBA_2", "DQN", "PPO", "LIQUID", "TFT")
@@ -75,6 +81,9 @@ message StartTrainingRequest {
bool use_gpu = 4; // Whether to use GPU acceleration
string description = 5; // Optional job description
map<string, string> tags = 6; // Optional categorization tags
TrainingMode mode = 7; // FULL (default) or FINE_TUNE
string resume_checkpoint_path = 8; // Path to checkpoint for fine-tune
uint32 max_epochs = 9; // Override epoch count (0 = use default)
}
message StartTrainingResponse {