diff options
Diffstat (limited to 'src/research.py')
| -rw-r--r-- | src/research.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/research.py b/src/research.py index f0374cf..d660f8e 100644 --- a/src/research.py +++ b/src/research.py @@ -37,6 +37,8 @@ class Research: self.plotter.plot(self.model) if preview: self.plotter.show() + if save_plot: + self.plotter.save(os.path.join(self.path, f'{iteration}.png')) if iteration % 50 == 0: if save_model: |