Commit 1142828
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -224,6 +224,10 @@ History FitInternal(DataHandler data_handler, int epochs, int validation_step, i | |||
| 224 | 224 | ||
| 225 | 225 | GC.Collect(); | |
| 226 | 226 | GC.WaitForPendingFinalizers(); | |
| 227 | + if (stop_training) | ||
| 228 | + { | ||
| 229 | + break; | ||
| 230 | + } | ||
| 227 | 231 | } | |
| 228 | 232 | ||
| 229 | 233 | return callbacks.History; | |
@@ -283,6 +287,10 @@ History FitInternal(DataHandler data_handler, int epochs, int verbose, List<ICal | |||
| 283 | 287 | ||
| 284 | 288 | GC.Collect(); | |
| 285 | 289 | GC.WaitForPendingFinalizers(); | |
| 290 | + if (stop_training) | ||
| 291 | + { | ||
| 292 | + break; | ||
| 293 | + } | ||
| 286 | 294 | } | |
| 287 | 295 | ||
| 288 | 296 | return callbacks.History; | |
@@ -339,6 +347,10 @@ History FitInternal(DataHandler data_handler, int epochs, int verbose, List<ICal | |||
| 339 | 347 | ||
| 340 | 348 | GC.Collect(); | |
| 341 | 349 | GC.WaitForPendingFinalizers(); | |
| 350 | + if (stop_training) | ||
| 351 | + { | ||
| 352 | + break; | ||
| 353 | + } | ||
| 342 | 354 | } | |
| 343 | 355 | ||
| 344 | 356 | return callbacks.History; | |
0 commit comments