diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-05-02 20:26:39 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-05-02 20:26:39 -0700 |
| commit | 349614a2f168654ba59bf1461ea61e1cb9358eb6 (patch) | |
| tree | be9829470eb5dddc673b9019f552a1292f2e3ecf /util/visualizer.py | |
| parent | e9146a659a896fc71ebf59cc57248449091e8d16 (diff) | |
use InstanceNorm2d from the latest pytorch
Diffstat (limited to 'util/visualizer.py')
| -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 9a6da2a..46348be 100644 --- a/util/visualizer.py +++ b/util/visualizer.py @@ -70,8 +70,8 @@ class Visualizer(): win=self.display_id) # errors: same format as |errors| of plotCurrentErrors - def print_current_errors(self, epoch, i, errors, start_time): - message = '(epoch: %d, iters: %d, time: %.3f) ' % (epoch, i, time.time() - start_time) + def print_current_errors(self, epoch, i, errors, t): + message = '(epoch: %d, iters: %d, time: %.3f) ' % (epoch, i, t) for k, v in errors.items(): message += '%s: %.3f ' % (k, v) |