From 7986185fcb778dbc2f04c5f690332603e4925cbe Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 1 Oct 2020 11:28:06 -0600 Subject: [PATCH] Change 'mod_step' to 'every' --- codes/models/ExtensibleTrainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codes/models/ExtensibleTrainer.py b/codes/models/ExtensibleTrainer.py index 649087c2..395f88bc 100644 --- a/codes/models/ExtensibleTrainer.py +++ b/codes/models/ExtensibleTrainer.py @@ -177,8 +177,8 @@ class ExtensibleTrainer(BaseModel): # Iterate through the steps, performing them one at a time. state = self.dstate for step_num, s in enumerate(self.steps): - # Skip steps if mod_step doesn't line up. - if 'mod_step' in s.step_opt.keys() and step % s.step_opt['mod_step'] != 0: + # 'every' is used to denote steps that should only occur at a certain integer factor rate. e.g. '2' occurs every 2 steps. + if 'every' in s.step_opt.keys() and step % s.step_opt['every'] != 0: continue # Steps can opt out of early (or late) training, make sure that happens here. if 'after' in s.step_opt.keys() and step < s.step_opt['after'] or 'before' in s.step_opt.keys() and step > s.step_opt['before']: