diff options
Diffstat (limited to 'tools/tensorflow_model_freezer/sample/Operation_gen.py')
-rw-r--r-- | tools/tensorflow_model_freezer/sample/Operation_gen.py | 20 |
1 files changed, 9 insertions, 11 deletions
diff --git a/tools/tensorflow_model_freezer/sample/Operation_gen.py b/tools/tensorflow_model_freezer/sample/Operation_gen.py index be4d74b2c..3a810e53d 100644 --- a/tools/tensorflow_model_freezer/sample/Operation_gen.py +++ b/tools/tensorflow_model_freezer/sample/Operation_gen.py @@ -27,7 +27,7 @@ import model_freezer_util as util # see MUL_gen.py for details usage and sample class GenFloor(base.BaseFreezer): def __init__(self, path): - super(self.__class__, self).__init__(path) + super(GenFloor, self).__init__(path) def getOutputDirectory(self): return os.path.join(self.root_output_path, @@ -66,7 +66,7 @@ class GenFloor(base.BaseFreezer): class GenPad(base.BaseFreezer): def __init__(self, path): - super(self.__class__, self).__init__(path) + super(GenPad, self).__init__(path) def getOutputDirectory(self): return os.path.join(self.root_output_path, @@ -111,7 +111,7 @@ class GenPad(base.BaseFreezer): class GenSqueeze(base.BaseFreezer): def __init__(self, path): - super(self.__class__, self).__init__(path) + super(GenSqueeze, self).__init__(path) def getOutputDirectory(self): return os.path.join(self.root_output_path, @@ -150,7 +150,7 @@ class GenSqueeze(base.BaseFreezer): class GenTranspose(base.BaseFreezer): def __init__(self, path): - super(self.__class__, self).__init__(path) + super(GenTranspose, self).__init__(path) def getOutputDirectory(self): return os.path.join(self.root_output_path, @@ -187,13 +187,11 @@ class GenTranspose(base.BaseFreezer): return (input_list, [output_node]) -''' -How to run -$ chmod +x tools/tensorflow_model_freezer/sample/name_of_this_file.py -$ PYTHONPATH=$PYTHONPATH:./tools/tensorflow_model_freezer/ \ - tools/tensorflow_model_freezer/sample/name_of_this_file.py \ - ~/temp # directory where model files are saved -''' +# How to run +# $ chmod +x tools/tensorflow_model_freezer/sample/name_of_this_file.py +# $ PYTHONPATH=$PYTHONPATH:./tools/tensorflow_model_freezer/ \ +# tools/tensorflow_model_freezer/sample/name_of_this_file.py \ +# ~/temp # directory where model files are saved # -------- if __name__ == "__main__": |