mirror of
https://github.com/zebrajr/faceswap.git
synced 2025-12-06 00:20:09 +01:00
Compare commits
2 Commits
b02075647b
...
fbbdf5522c
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
fbbdf5522c | ||
|
|
b16b755d55 |
|
|
@ -132,7 +132,8 @@ class LearningRateFinder:
|
|||
for idx in pbar:
|
||||
model_inputs, model_targets = self._feeder.get_batch()
|
||||
loss: list[float] = self._model.model.train_on_batch(model_inputs, y=model_targets)
|
||||
if np.isnan(loss[0]):
|
||||
if any(np.isnan(x) for x in loss):
|
||||
logger.warning("NaN detected! Exiting early")
|
||||
break
|
||||
self._on_batch_end(idx, loss[0])
|
||||
self._update_description(pbar)
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user