diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 18:05:13 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 18:05:13 +0100 |
| commit | 36bcb2d89e38f6e89d6a540649654fa2d921f832 (patch) | |
| tree | f4303a620c94bdc6e3d0679933722751d41ba9fb /inversion | |
| parent | 786766986deac14e547f484e8911a93034171c81 (diff) | |
break
Diffstat (limited to 'inversion')
| -rw-r--r-- | inversion/image_inversion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/inversion/image_inversion.py b/inversion/image_inversion.py index 158b7e9..da43b6b 100644 --- a/inversion/image_inversion.py +++ b/inversion/image_inversion.py @@ -479,7 +479,7 @@ for image_batch, label_batch in image_gen: out_labels[out_pos:out_pos+BATCH_SIZE] = label_batch out_err[out_pos:out_pos+BATCH_SIZE] = rec_err_batch out_pos += BATCH_SIZE - if out_pos >= params.max_batches: + if (out_pos / BATCH_SIZE) >= params.max_batches: break print('Mean reconstruction error: {}'.format(np.mean(out_err))) |