Fix/cache callback reset data (#570)

* Fix bug where cache imaging callback didn't reset data * Update changelog
pytorchbearer · Jun 13, 2019 · 86389d2 · 86389d2
1 parent 068f84c
commit 86389d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -15,6 +15,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 ### Fixed
 - Fixed a bug where list or dictionary metrics would cause the tensorboard callback to error
 - Fixed a bug where running a trial without training steps would error
+- Fixed a bug where the caching imaging callback didn't reset data so couldn't be run in multiple trials
 
 ## [0.3.2] - 2019-05-28
 ### Added

diff --git a/torchbearer/callbacks/imaging/imaging.py b/torchbearer/callbacks/imaging/imaging.py
@@ -249,6 +249,7 @@ def on_batch(self, state):
 
     def on_end_epoch(self, state):
         super(CachingImagingCallback, self).on_end_epoch(state)
+        self._data = None
         self._done = False