From 0534e85981b327b8c0effb9b0814ae041a4fb905 Mon Sep 17 00:00:00 2001
From: eug-vs <eugene@eug-vs.xyz>
Date: Mon, 23 May 2022 11:59:05 +0400
Subject: refactor: use iterator for grid

---
 src/research.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src/research.py')

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:
-- 
cgit v1.2.3