coosld/tig-monorepo
1
0
Fork 0
mirror of https://github.com/tig-foundation/tig-monorepo.git synced 2026-02-21 10:27:49 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Save every epoch where loss improves.
Some checks failed
Test Workspace / Test Workspace (push) Has been cancelled
Details

Browse Source
This commit is contained in:
FiveMovesAhead 2025-12-03 14:11:08 +00:00
parent c4f12d7350
commit 0a12ff6fe1
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tig-challenges/src/neuralnet_optimizer/mod.rs
View File

@ -381,7 +381,6 @@ pub fn training_loop(
)?;
let mut lowest_loss = f32::INFINITY;
let mut last_save_epoch = None;
let mut epochs_no_improvement = 0;
let mut prev_train_loss = None;
let mut prev_validation_loss = None;
@ -568,17 +567,10 @@ pub fn training_loop(
if avg_val_loss < lowest_loss - min_loss_delta {
lowest_loss = avg_val_loss;
epochs_no_improvement = 0;
if !last_save_epoch
.as_ref()
.is_some_and(|&x| epoch - x < max_epochs / 20)
{
last_save_epoch = Some(epoch);
save_solution(&to_solution(&model, epoch + 1, stream.clone())?)?;
}
save_solution(&to_solution(&model, epoch + 1, stream.clone())?)?;
} else {
epochs_no_improvement += 1;
if epochs_no_improvement >= patience {
save_solution(&to_solution(&model, epoch + 1, stream.clone())?)?;
break;
}
}