diff options
| author | junyanz <junyanz@mit.edu> | 2018-02-18 22:53:26 -0500 |
|---|---|---|
| committer | junyanz <junyanz@mit.edu> | 2018-02-18 22:53:26 -0500 |
| commit | 079da5c02fd99ef35d7cad0e20c2924b7c2bcffd (patch) | |
| tree | 8d00061c5b639f0ad832ab2a34a76d10a076dff7 /util | |
| parent | 51bd910bc737984163f6d6534cccbd22668e2b28 (diff) | |
fix test_model & add timer for data loader & rename identity loss
Diffstat (limited to 'util')
| -rw-r--r-- | util/visualizer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/util/visualizer.py b/util/visualizer.py index b22f235..fd8140d 100644 --- a/util/visualizer.py +++ b/util/visualizer.py @@ -114,8 +114,8 @@ class Visualizer(): win=self.display_id) # errors: same format as |errors| of plotCurrentErrors - def print_current_errors(self, epoch, i, errors, t): - message = '(epoch: %d, iters: %d, time: %.3f) ' % (epoch, i, t) + def print_current_errors(self, epoch, i, errors, t, t_data): + message = '(epoch: %d, iters: %d, time: %.3f, data: %.3f) ' % (epoch, i, t, t_data) for k, v in errors.items(): message += '%s: %.3f ' % (k, v) |