CameronJGrant/strafe-ai
1
Fork 0
forked from StrafesNET/strafe-ai
Code Pull Requests Activity

write model name based on num params

Browse Source
This commit is contained in:
Rhys Lloyd
2026-03-27 16:16:26 -07:00
parent 03f5eb5c13
commit 7bf439395b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/training.rs
View File

@@ -160,7 +160,8 @@ fn training(gpu_id: usize, epochs: usize, learning_rate: f64) {
let device = burn::backend::cuda::CudaDevice::new(gpu_id);
let mut model: Net<TrainingBackend> = Net::init(&device);
println!("Training model ({} parameters)", model.num_params());
let num_params = model.num_params();
println!("Training model ({} parameters)", num_params);
let mut optim = AdamConfig::new().init();
@@ -205,7 +206,7 @@ fn training(gpu_id: usize, epochs: usize, learning_rate: f64) {
}
}
let date_string = format!("{}_{}.model", chrono::Utc::now(), best_loss);
let date_string = format!("{}_{}.model", num_params, best_loss);
best_model
.save_file(
date_string,