NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix running gradient crashing sometimes (#2401)

Browse Source
This commit is contained in:
Adrià Arrufat 2021-08-04 19:59:42 +09:00 committed by GitHub
parent 16500906b0
commit bec25d8247
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dlib/dnn/trainer.h
View File

@ -1121,7 +1121,7 @@ namespace dlib
}
// if we haven't seen much data yet then just say false.
if (gradient_updates_since_last_sync < 30)
if (previous_loss_values_to_keep_until_disk_sync.size() < 30)
return false;
// if the loss is very likely to be increasing then return true