From 336e0ca944ba170c43255fbf116cd8ae57114f8c Mon Sep 17 00:00:00 2001 From: iperov Date: Sat, 17 Jul 2021 22:04:20 +0400 Subject: [PATCH] fix --- models/ModelBase.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/models/ModelBase.py b/models/ModelBase.py index 598c19a..f446efa 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -22,6 +22,7 @@ from samplelib import SampleGeneratorBase class ModelBase(object): def __init__(self, is_training=False, + is_exporting=False, saved_models_path=None, training_data_src_path=None, training_data_dst_path=None, @@ -36,6 +37,7 @@ class ModelBase(object): silent_start=False, **kwargs): self.is_training = is_training + self.is_exporting = is_exporting self.saved_models_path = saved_models_path self.training_data_src_path = training_data_src_path self.training_data_dst_path = training_data_dst_path