diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 18:04:30 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 18:04:30 +0100 |
| commit | 2b69f55f36a6ce4b68189f3063cc08eea33738e5 (patch) | |
| tree | 9e07d0e4d66e510ad02e8966e3b1cad2864229b2 /inversion/image_inversion.py | |
| parent | 2a9fbde05922ba0a580546c822942ca77de00e85 (diff) | |
break
Diffstat (limited to 'inversion/image_inversion.py')
| -rw-r--r-- | inversion/image_inversion.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/inversion/image_inversion.py b/inversion/image_inversion.py index 274353b..db2bd02 100644 --- a/inversion/image_inversion.py +++ b/inversion/image_inversion.py @@ -479,6 +479,8 @@ 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: + break print('Mean reconstruction error: {}'.format(np.mean(out_err))) print('Stdev reconstruction error: {}'.format(np.std(out_err))) |